Update TEED/main.py

TEED/main.py

@@ -366,7 +366,7 @@ def main(args, train_inf):
     training_dir = os.path.join(args.output_dir,args.train_data)
     os.makedirs(training_dir,exist_ok=True)
     checkpoint_path = './TEED/checkpoints/BIPED/5/5_model.pth'
-    checkpoint_path = os.path.join('./TEED/checkpoints', args.dataset, str(args.epoch), '5_model.pth')
+    checkpoint_path = os.path.join('./TEED/checkpoints', 'BIPED', str(args.epoch), '5_model.pth')
     if args.tensorboard and not args.is_testing:
         # from tensorboardX import SummaryWriter  # previous torch version
         from torch.utils.tensorboard import SummaryWriter # for torch 1.4 or greather