update

train_multigpu.py

@@ -31,9 +31,9 @@ def getParameters():
     
     parser = argparse.ArgumentParser()
     # general settings
-    parser.add_argument('-v', '--version', type=str, default='ori_tiny',
+    parser.add_argument('-v', '--version', type=str, default='cycle_lstu1',
                                             help="version name for train, test, finetune")
-    parser.add_argument('-t', '--tag', type=str, default='tiny',
+    parser.add_argument('-t', '--tag', type=str, default='cycle',
                                             help="tag for current experiment")
 
     parser.add_argument('-p', '--phase', type=str, default="train",
@@ -46,9 +46,9 @@ def getParameters():
 
     # training
     parser.add_argument('--experiment_description', type=str,
-                                default="只用conv，训练最小的模型")
+                                default="cycle配合LSTU")
 
-    parser.add_argument('--train_yaml', type=str, default="train_ori_modulation_config.yaml")
+    parser.add_argument('--train_yaml', type=str, default="train_cycleloss.yaml")
 
     # system logger
     parser.add_argument('--logger', type=str,