Spaces:

RashiAgarwal
/

YOLO_v3_PyTorchLightning

Paused

RashiAgarwal commited on Aug 20, 2023

Commit

aa041fe

•

1 Parent(s): 2ce5d17

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(