small updates

2023-05-22 00:14:18 +02:00
parent 52e461ca47
commit 6b81c0d1a6
5 changed files with 202 additions and 7 deletions
@@ -13,7 +13,8 @@ args = vars(parser.parse_args())

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