From d4ff8a06fe86f3f84eca895a2205cd1c8799655a Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Tue, 23 Mar 2021 01:06:24 -0500
Subject: [PATCH] More torch2hpvm README Path fixes
---
hpvm/projects/torch2hpvm/README.md | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/hpvm/projects/torch2hpvm/README.md b/hpvm/projects/torch2hpvm/README.md
index 493ad26447..1f06142f52 100644
--- a/hpvm/projects/torch2hpvm/README.md
+++ b/hpvm/projects/torch2hpvm/README.md
@@ -26,9 +26,9 @@ First, prepare 2 datasets for autotuning and testing.
```python
from torch2hpvm import BinDataset
-from pathlib import Path, PosixPath
+from pathlib import Path
-data_dir = Path(__file__).parent / "model_params/resnet18_cifar10"
+data_dir = Path(__file__).parent / "../model_params/resnet18_cifar10"
dataset_shape = 5000, 3, 32, 32
tuneset = BinDataset(data_dir / "tune_input.bin", data_dir / "tune_labels.bin", dataset_shape)
testset = BinDataset(data_dir / "test_input.bin", data_dir / "test_labels.bin", dataset_shape)
@@ -52,7 +52,7 @@ from torch.nn import Module
import dnn # Defined at `hpvm/test/dnn_benchmarks/pytorch`
model: Module = dnn.ResNet18()
-checkpoint = Path(__file__).parent.absolute() / PosixPath("../model_params/resnet18_cifar10.pth.tar")
+checkpoint = Path(__file__).parent / "../model_params/resnet18_cifar10.pth.tar"
model.load_state_dict(torch.load(checkpoint))
```
@@ -108,3 +108,4 @@ This choice of operators is largely constrained by backend (tensor_runtime) supp
1. Expand the list of operators supported in the frontend.
- Most ideally, create a high-level description of operators that can tie
HPVM-C intrinsics and the frontend list of operators together.
+
--
GitLab