add some configs

megvii-research · Jul 2, 2020 · 8f951b4 · 8f951b4
1 parent c9ec788
commit 8f951b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/backbone.py b/backbone.py
@@ -3,7 +3,7 @@
 
 class ResNet12Block(nn.Module):
     """
-    ResNet block
+    ResNet Block
     """
     def __init__(self, inplanes, planes):
         super(ResNet12Block, self).__init__()
@@ -45,7 +45,7 @@ def forward(self, x):
 
 class ResNet12(nn.Module):
     """
-    ResNet12 backbone
+    ResNet12 Backbone
     """
     def __init__(self, emb_size, block=ResNet12Block, cifar_flag=False):
         super(ResNet12, self).__init__()
@@ -116,7 +116,7 @@ def forward(self, x):
 
 class ConvNet(nn.Module):
     """
-    Conv4 backbone
+    Conv4 Backbone
     """
     def __init__(self, emb_size, cifar_flag=False):
         super(ConvNet, self).__init__()

diff --git a/main.py b/main.py
@@ -490,7 +490,7 @@ def main():
         print('Backbone: ConvNet')
     else:
         logger.info('Invalid backbone: {}, please specify a backbone model from '
-                    'convnet and resnet12.'.format(config['backbone']))
+                    'convnet or resnet12.'.format(config['backbone']))
         exit()
 
     gnn_module = DPGN(config['num_generation'],