diff --git a/vitookit/evaluation/eval_cls_ffcv.py b/vitookit/evaluation/eval_cls_ffcv.py
index 70a79f0645280dda6fffc0507d2f83c60317ed7b..91d3e323095f377d5b53dfdcadac4509288e980d 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 143b6a8f7fa58120aa56fad70d9265dbdeb781bf..e5bcb20e47179877aba8b86e8d183885a708cda2 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