Gosula commited on
Commit
2715392
1 Parent(s): 4c1bb98

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -13,7 +13,7 @@ List is structured by "B" indicating a residual block followed by the number of
13
  "S" is for scale prediction block and computing the yolo loss
14
  "U" is for upsampling the feature map and concatenating with a previous layer
15
  """
16
- config = [
17
  (32, 3, 1),
18
  (64, 3, 2),
19
  ["B", 1],
@@ -130,7 +130,7 @@ class YOLOv3(LightningModule):
130
  layers = nn.ModuleList()
131
  in_channels = self.in_channels
132
 
133
- for module in config:
134
  if isinstance(module, tuple):
135
  out_channels, kernel_size, stride = module
136
  layers.append(
 
13
  "S" is for scale prediction block and computing the yolo loss
14
  "U" is for upsampling the feature map and concatenating with a previous layer
15
  """
16
+ config_1 = [
17
  (32, 3, 1),
18
  (64, 3, 2),
19
  ["B", 1],
 
130
  layers = nn.ModuleList()
131
  in_channels = self.in_channels
132
 
133
+ for module in config_1:
134
  if isinstance(module, tuple):
135
  out_channels, kernel_size, stride = module
136
  layers.append(