add naswot

graph_dit/naswot/config_utils/share_args.py

@@ -0,0 +1,17 @@
+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')