update training script

GestaltCogTeam · Oct 31, 2023 · a9a2739 · a9a2739
1 parent 66f7d33
commit a9a2739
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/experiments/train.py b/experiments/train.py
@@ -15,9 +15,8 @@
 
 def parse_args():
     parser = ArgumentParser(description="Run time series forecasting model in BasicTS framework!")
-    parser.add_argument("-c", "--cfg", default="baselines/DGCRN/DGCRN_METR-LA.py", help="training config")
-    # parser.add_argument("-c", "--cfg", default="baselines/STGODE/STGODE_PEMS08.py", help="training config")
-    parser.add_argument("-g", "--gpus", default="3", help="visible gpus")
+    parser.add_argument("-c", "--cfg", default="baselines/STID/METR-LA.py", help="training config")
+    parser.add_argument("-g", "--gpus", default="0", help="visible gpus")
     return parser.parse_args()
 
 if __name__ == "__main__":