Spaces:

LN1996
/

S13-ERA-Phase-I-Yolov3-Pascal

Sleeping

LN1996 commited on Aug 18, 2023

Commit

25a0da8

1 Parent(s): 522ddae

Update loss.py

Files changed (1) hide show

loss.py CHANGED Viewed

@@ -6,11 +6,10 @@ instead of BinaryCrossEntropy.
 import random
 import torch
 import torch.nn as nn
-import lightning.pytorch as pl
 from utils import intersection_over_union
-class YoloLoss(pl.LightningModule):
     def __init__(self):
         super().__init__()
         self.mse = nn.MSELoss()

 import random
 import torch
 import torch.nn as nn
 from utils import intersection_over_union
+class YoloLoss(nn.Module):
     def __init__(self):
         super().__init__()
         self.mse = nn.MSELoss()