From 798d8a74e74f22842cfcbec8ac0180bd291777ff Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 25 Mar 2021 04:29:51 -0500
Subject: [PATCH] Fix up path to pytorch weight
---
hpvm/test/dnn_benchmarks/pytorch/test_frontend.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py b/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py
index 19f1736645..2fb1de17ee 100644
--- a/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py
+++ b/hpvm/test/dnn_benchmarks/pytorch/test_frontend.py
@@ -39,7 +39,7 @@ for model_cls, nch, img_size, batch_size, pathname in benchmarks:
params / "test_input.bin", params / "test_labels.bin", dataset_shape
)
model: Module = model_cls()
- checkpoint = self_folder / "../model_params" / f"{pathname}.pth.tar"
+ checkpoint = self_folder / "../model_params/pytorch" / f"{pathname}.pth.tar"
model.load_state_dict(torch.load(checkpoint.as_posix()))
build_dir = codegen_dir / "build"
--
GitLab