From 5811961949643892fc80ea3a5fd7a36f4b23f842 Mon Sep 17 00:00:00 2001
From: nullptr <nullptr@localhost>
Date: Tue, 11 Jul 2023 05:52:51 +0000
Subject: [PATCH] fix: use val_dataloader for tflite and onnx inference

---
 tools/inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/inference.py b/tools/inference.py
index 80652705..b4ecf20b 100644
--- a/tools/inference.py
+++ b/tools/inference.py
@@ -317,7 +317,7 @@ def after_test_epoch(self, _, metrics=None):
         # TODO: Support '.pickel' dump
         runner = Infernce(
             args.checkpoint,
-            dataloader=runner.test_dataloader,
+            dataloader=runner.val_dataloader,
             cfg=cfg,
             runner=runner,
             source=args.source,