Spaces:

NishantD
/

Segmentation

Sleeping

NishantD commited on Jun 9, 2024

Commit

4cf9e1d

verified ·

1 Parent(s): 375c042

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -1,5 +1,5 @@
 import os
-import lightning as L
 from dataloader import AerialImageDataset
 from train import UNet
 from torch.utils.data import DataLoader

 import os
+import pytorch_lightning as L
 from dataloader import AerialImageDataset
 from train import UNet
 from torch.utils.data import DataLoader