Move to xautodl

xautodl/config_utils/share_args.py

@@ -0,0 +1,39 @@
+import os, sys, time, random, argparse
+
+
+def add_shared_args(parser):
+    # Data Generation
+    parser.add_argument("--dataset", type=str, help="The dataset name.")
+    parser.add_argument("--data_path", type=str, help="The dataset name.")
+    parser.add_argument(
+        "--cutout_length", type=int, help="The cutout length, negative means not use."
+    )
+    # Printing
+    parser.add_argument(
+        "--print_freq", type=int, default=100, help="print frequency (default: 200)"
+    )
+    parser.add_argument(
+        "--print_freq_eval",
+        type=int,
+        default=100,
+        help="print frequency (default: 200)",
+    )
+    # Checkpoints
+    parser.add_argument(
+        "--eval_frequency",
+        type=int,
+        default=1,
+        help="evaluation frequency (default: 200)",
+    )
+    parser.add_argument(
+        "--save_dir", type=str, help="Folder to save checkpoints and log."
+    )
+    # Acceleration
+    parser.add_argument(
+        "--workers",
+        type=int,
+        default=8,
+        help="number of data loading workers (default: 8)",
+    )
+    # Random Seed
+    parser.add_argument("--rand_seed", type=int, default=-1, help="manual seed")