Spaces:

RashiAgarwal
/

YOLO_v3_PyTorchLightning

Paused

RashiAgarwal commited on Aug 20, 2023

Commit

d55a993

•

1 Parent(s): 3b48481

Update yolov3.py

Files changed (1) hide show

yolov3.py CHANGED Viewed

@@ -17,7 +17,7 @@ class YOLOV3_PL(LightningModule):
         self.learning_rate = learning_rate
         self.num_epochs = num_epochs
         self.scaled_anchors = config.SCALED_ANCHORS
-        #self.layers = self.model.layers
     def train_dataloader(self):
         self.train_data = YOLODataset(

         self.learning_rate = learning_rate
         self.num_epochs = num_epochs
         self.scaled_anchors = config.SCALED_ANCHORS
+        self.layers = self.model.layers
     def train_dataloader(self):
         self.train_data = YOLODataset(