From bd711be6530f9e85b3a3b3d5c48ab9eabb4459a8 Mon Sep 17 00:00:00 2001
From: gent <jw02425@surrey.ac.uk>
Date: Thu, 18 Jan 2024 21:46:27 +0000
Subject: [PATCH] Fix imports and remove unused code
---
vitookit/evaluation/eval_cls_ffcv.py | 6 ++----
vitookit/evaluation/eval_linear_ffcv.py | 4 ++--
2 files changed, 4 insertions(+), 6 deletions(-)
diff --git a/vitookit/evaluation/eval_cls_ffcv.py b/vitookit/evaluation/eval_cls_ffcv.py
index 70a79f0..91d3e32 100644
--- a/vitookit/evaluation/eval_cls_ffcv.py
+++ b/vitookit/evaluation/eval_cls_ffcv.py
@@ -30,20 +30,18 @@ from vitookit.datasets.transform import ValPipeline, three_augmentation
from vitookit.utils.helper import *
from vitookit.utils import misc
from vitookit.models.build_model import build_model
-from vitookit.datasets import build_dataset
import wandb
from pathlib import Path
from typing import Iterable, Optional
from torch.nn import functional as F
-from timm.models import create_model
+
from timm.loss import LabelSmoothingCrossEntropy, SoftTargetCrossEntropy
from timm.scheduler import create_scheduler
from timm.optim import create_optimizer
from timm.utils import NativeScaler, get_state_dict, ModelEma, accuracy
-from timm.data import Mixup, create_transform
-from timm.data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
+from timm.data import Mixup
from timm.layers import trunc_normal_
from ffcv import Loader
diff --git a/vitookit/evaluation/eval_linear_ffcv.py b/vitookit/evaluation/eval_linear_ffcv.py
index 143b6a8..e5bcb20 100644
--- a/vitookit/evaluation/eval_linear_ffcv.py
+++ b/vitookit/evaluation/eval_linear_ffcv.py
@@ -19,7 +19,7 @@ import os
import time
from pathlib import Path
-import torch
+
import torch.backends.cudnn as cudnn
import wandb
from vitookit.models.build_model import build_model
@@ -126,7 +126,7 @@ def main(args):
print('job dir: {}'.format(os.path.dirname(os.path.realpath(__file__))))
print("{}".format(args).replace(', ', ',\n'))
-
+ import torch
device = torch.device(args.device)
# fix the seed for reproducibility
--
GitLab