fix: use val_dataloader for tflite and onnx inference

Seeed-Studio · Jul 11, 2023 · 5811961 · 5811961
1 parent 7a869bb
commit 5811961
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,