From f07c128d9db5603fabb8ea07f5d42c3f2a2e0b54 Mon Sep 17 00:00:00 2001
From: gent <jw02425@surrey.ac.uk>
Date: Thu, 18 Jan 2024 21:07:34 +0000
Subject: [PATCH] Import torch module and fix seed for reproducibility

---
 vitookit/evaluation/eval_linear_ffcv.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/vitookit/evaluation/eval_linear_ffcv.py b/vitookit/evaluation/eval_linear_ffcv.py
index b37c079..26cabe0 100644
--- a/vitookit/evaluation/eval_linear_ffcv.py
+++ b/vitookit/evaluation/eval_linear_ffcv.py
@@ -25,9 +25,7 @@ import wandb
 from vitookit.models.build_model import build_model
 
 from vitookit.utils import misc
-from vitookit.datasets import build_dataset
 
-from torchvision.transforms import *
 from vitookit.utils.helper import aug_parse, load_pretrained_weights, restart_from_checkpoint
 from timm.models.layers import trunc_normal_
 
@@ -128,6 +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