trying skip connection

2023-05-24 17:46:13 +02:00
parent 53217c6e5d
commit 5a1276f045
4 changed files with 121 additions and 24 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.direct_1 import config
+    from configs.direct_3 import config
    # from configs.ae_static_orig_2 import config
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']