diff --git a/hpvm/projects/torch2hpvm/README.md b/hpvm/projects/torch2hpvm/README.md index 1e013e936aa8c5a2d4d492d8096be6f3c783b6d8..493ad26447b60afbd6e76885722dc7d9976e2fd7 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)) ```