From 2d274ce239db8a4323be4160262c2977e92ec4b8 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Tue, 23 Mar 2021 00:33:03 -0500
Subject: [PATCH] PosixPath fixes to code snippets
---
hpvm/projects/torch2hpvm/README.md | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/hpvm/projects/torch2hpvm/README.md b/hpvm/projects/torch2hpvm/README.md
index 1e013e936a..493ad26447 100644
--- a/hpvm/projects/torch2hpvm/README.md
+++ b/hpvm/projects/torch2hpvm/README.md
@@ -26,7 +26,7 @@ First, prepare 2 datasets for autotuning and testing.
```python
from torch2hpvm import BinDataset
-from pathlib import Path
+from pathlib import Path, PosixPath
data_dir = Path(__file__).parent / "model_params/resnet18_cifar10"
dataset_shape = 5000, 3, 32, 32
@@ -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 / "model_params/resnet18_cifar10.pth.tar"
+checkpoint = Path(__file__).parent.absolute() / PosixPath("../model_params/resnet18_cifar10.pth.tar")
model.load_state_dict(torch.load(checkpoint))
```
--
GitLab