cleanup classif

Author: NicolasHug

references/classification/presets.py

@@ -1,5 +1,5 @@
 import torch
-from torchvision.prototype import transforms
+import torchvision.transforms.v2 as transforms
 
 
 class ClassificationPresetTrain:

references/classification/train.py

@@ -14,7 +14,7 @@
 import transforms
 from torch import nn
 from torch.utils.data.dataloader import default_collate
-from torchvision.prototype.transforms import InterpolationMode
+from torchvision.transforms.v2 import InterpolationMode
 
 
 def train_one_epoch(model, criterion, optimizer, data_loader, device, epoch, args, model_ema=None, scaler=None):

references/classification/transforms.py

@@ -3,8 +3,6 @@
 
 import torch
 from torch import Tensor
-from torchvision.prototype import features
-from torchvision.prototype.transforms import functional as PF
 from torchvision.transforms import functional as F