From da6e8daaf2c7c16cea127a77725f0ef5334fad9d Mon Sep 17 00:00:00 2001
From: gent <jw02425@surrey.ac.uk>
Date: Sun, 3 Mar 2024 14:43:55 +0000
Subject: [PATCH] Add DynamicResolution dataset import
---
vitookit/evaluation/eval_cls_ffcv.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/vitookit/evaluation/eval_cls_ffcv.py b/vitookit/evaluation/eval_cls_ffcv.py
index 2450174..c737aa1 100644
--- a/vitookit/evaluation/eval_cls_ffcv.py
+++ b/vitookit/evaluation/eval_cls_ffcv.py
@@ -27,6 +27,7 @@ from vitookit.evaluation.eval_cls import evaluate
from vitookit.utils.helper import *
from vitookit.utils import misc
from vitookit.models.build_model import build_model
+from vitookit.datasets.dres import DynamicResolution
import wandb
from pathlib import Path
@@ -43,7 +44,6 @@ from timm.layers import trunc_normal_
from ffcv import Loader
from ffcv.loader import OrderOption
-from datasets.dres import DynamicResolution
def get_args_parser():
parser = argparse.ArgumentParser('DeiT training and evaluation script', add_help=False)
--
GitLab