Gosula commited on
Commit
4f17bbc
1 Parent(s): 589280a

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -4,7 +4,7 @@ Implementation of YOLOv3 architecture
4
  from pytorch_lightning import LightningModule
5
  import torch
6
  import torch.nn as nn
7
-
8
  """
9
  Information about architecture config:
10
  Tuple is structured by (filters, kernel_size, stride)
 
4
  from pytorch_lightning import LightningModule
5
  import torch
6
  import torch.nn as nn
7
+ import config
8
  """
9
  Information about architecture config:
10
  Tuple is structured by (filters, kernel_size, stride)