Add files via upload

Herrykite · Dec 18, 2020 · 5d6a197 · 5d6a197
1 parent 72b1f44
commit 5d6a197
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modeling/cnn.py b/modeling/cnn.py
@@ -2,11 +2,13 @@
 import sys
 
 sys.path.insert(0, '../../')
+import os
 from collections import OrderedDict
 from torch import nn
 from ConvNet.config.defaults import get_cfg_defaults
 
 cfg = get_cfg_defaults()
+cfg.merge_from_file(cfg.MODEL.CONFIG + os.listdir(cfg.MODEL.CONFIG)[-1])
 
 
 class PCAnet(nn.Module):