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