update cape norm training

2023-07-01 10:57:55 +02:00
parent cac04a1610
commit 05fc297808
3 changed files with 152 additions and 3 deletions
@@ -11,10 +11,12 @@ parser.add_argument('--config', type=str)
 parser.add_argument('--debug', action='store_true')
 args = vars(parser.parse_args())

-config_module = importlib.import_module(args['config'])
-config = config_module.config
 if args['debug']:
-    from configs.ae_norm_1 import config
+    from configs.iccvw.ae_norm_cape_m import config
+else:
+    config_module = importlib.import_module(args['config'])
+    config = config_module.config
+    
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 if args['debug']: