Spaces:
Sleeping
Sleeping
Update resnet.py
Browse files
resnet.py
CHANGED
|
@@ -22,8 +22,6 @@ from torchvision.datasets import CIFAR10
|
|
| 22 |
# from pytorch_lightning.callbacks import ModelSummary
|
| 23 |
# from lightning.pytorch.callbacks import ModelCheckpoint
|
| 24 |
from pytorch_lightning.callbacks import ModelCheckpoint, ModelSummary
|
| 25 |
-
import albumentations as A
|
| 26 |
-
from albumentations.pytorch import ToTensorV2
|
| 27 |
import torchvision.transforms as transforms
|
| 28 |
|
| 29 |
|
|
|
|
| 22 |
# from pytorch_lightning.callbacks import ModelSummary
|
| 23 |
# from lightning.pytorch.callbacks import ModelCheckpoint
|
| 24 |
from pytorch_lightning.callbacks import ModelCheckpoint, ModelSummary
|
|
|
|
|
|
|
| 25 |
import torchvision.transforms as transforms
|
| 26 |
|
| 27 |
|