Commit
•
32d95bb
1
Parent(s):
e80b224
Update yolov3.py
Browse files
yolov3.py
CHANGED
|
@@ -17,6 +17,7 @@ class YOLOV3_PL(LightningModule):
|
|
| 17 |
self.learning_rate = learning_rate
|
| 18 |
self.num_epochs = num_epochs
|
| 19 |
self.scaled_anchors = config.SCALED_ANCHORS
|
|
|
|
| 20 |
|
| 21 |
def train_dataloader(self):
|
| 22 |
self.train_data = YOLODataset(
|
|
|
|
| 17 |
self.learning_rate = learning_rate
|
| 18 |
self.num_epochs = num_epochs
|
| 19 |
self.scaled_anchors = config.SCALED_ANCHORS
|
| 20 |
+
self.layers = self.model.layers
|
| 21 |
|
| 22 |
def train_dataloader(self):
|
| 23 |
self.train_data = YOLODataset(
|