MamaPearl
/

nula-cifar10-robust-v0

Image Classification

computer-vision

adversarial-robustness

Model card Files Files and versions

MamaPearl commited on Apr 1

Commit

7a459ea

·

verified ·

1 Parent(s): 32f6a03

Update train_robust.py

Files changed (1) hide show

train_robust.py +1 -1

train_robust.py CHANGED Viewed

@@ -5,7 +5,7 @@ from tqdm.auto import tqdm
 from torch.optim.lr_scheduler import LinearLR, CosineAnnealingLR, SequentialLR
 from configuration_nula import NulaConfig
 from modeling_nula import NulaForImageClassification, BlurPool2d
-from nula_dataset import get_loaders, get_device
 from augmentations import resize_down_up, decimate
 NUM_EPOCHS = 50

 from torch.optim.lr_scheduler import LinearLR, CosineAnnealingLR, SequentialLR
 from configuration_nula import NulaConfig
 from modeling_nula import NulaForImageClassification, BlurPool2d
+from dataset_nula import get_loaders, get_device
 from augmentations import resize_down_up, decimate
 NUM_EPOCHS = 50