Update TEED/main.py

TEED/main.py

@@ -365,8 +365,8 @@ def main(args, train_inf):
     tb_writer = None
     training_dir = os.path.join(args.output_dir,args.train_data)
     os.makedirs(training_dir,exist_ok=True)
-   checkpoint_path = os.path.join('./teed', args.output_dir)
-checkpoint_path = os.path.join(checkpoint_path, args.train_data, args.checkpoint_data)
+    checkpoint_path = os.path.join('./teed', args.output_dir)
+    checkpoint_path = os.path.join(checkpoint_path, args.train_data, args.checkpoint_data)
     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