Update train_mr.py

showlab · Aug 21, 2023 · ec27ff9 · ec27ff9
1 parent 7dd381b
commit ec27ff9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/main/train_mr.py b/main/train_mr.py
@@ -15,7 +15,6 @@
 from torch.utils.data import DataLoader
 from torch.utils.tensorboard import SummaryWriter
 
-sys.path.append('/data/home/qinghonglin/univtg')
 from main.config import BaseOptions, setup_model
 from main.dataset import \
     DatasetMR, start_end_collate_mr, prepare_batch_inputs_mr
@@ -263,4 +262,4 @@ def start_training():
         logger.info("\n\n\nFINISHED TRAINING!!!")
         logger.info("Evaluating model at {}".format(best_ckpt_path))
         logger.info("Input args {}".format(sys.argv[1:]))
-        start_inference()
+        start_inference()