add

Ainimal · Nov 10, 2022 · 9566e5d · 9566e5d
1 parent 096be86
commit 9566e5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wama_modules/BaseModule.py b/wama_modules/BaseModule.py
@@ -350,7 +350,7 @@ class ResBlock(nn.Module):
  x_in → conv3x3 → norm → active → conv3x3 → norm → active → x_out
  ↘--→ conv1x1 if in_c != out_c) ---↗
 
- type1: '131', called BasicBlock, used for ResNet 50 and 101 and 152
+ type1: '131', called BottleNeck, used for ResNet 50 and 101 and 152
  x_in → conv3x3 → norm → active → conv3x3 → norm → active → x_out
  ↘--→ conv1x1 if in_c != out_c) ---↗
  """