pyramid ready for cape

2023-05-22 12:44:43 +02:00
parent 6b81c0d1a6
commit 60844f8011
5 changed files with 321 additions and 9 deletions
@@ -13,7 +13,7 @@ args = vars(parser.parse_args())
 config_module = importlib.import_module(args['config'])
 config = config_module.config
 if args['debug']:
-    from configs.pyr_32 import config
+    from configs.pyr_64_cape_m import config
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 if args['debug']: