Added changes to the main file

src/templates/template-vision-classification/main.py

@@ -11,8 +11,9 @@
 from torch import nn, optim
 from trainers import setup_evaluator, setup_trainer
 from utils import *
+#::: if (['fire'].includes(it.argparser)) { :::#
 import fire
-
+#::: } :::#
 
 def run(local_rank: int, config: Any):
  # make a certain seed