Skip to content

Commit

Permalink
fix model path
Browse files Browse the repository at this point in the history
Signed-off-by: chensuyue <[email protected]>
  • Loading branch information
chensuyue committed Jul 10, 2024
1 parent 7e83c0c commit 012081a
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion test/3x/torch/algorithms/fp8_quant/test_basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,10 @@ def forward(self, x):


model = Net()
checkpoint = torch.load("mnist-epoch_20.pth")
model_link = "https://vault.habana.ai/artifactory/misc/inference/mnist/mnist-epoch_20.pth"
model_path = "/tmp/.neural_compressor/mnist-epoch_20.pth"
os.system("mkdir -p /tmp/.neural_compressor && wget {} -O {} ".format(model_link, model_path))
checkpoint = torch.load(model_path)
model.load_state_dict(checkpoint)

model = model.eval()
Expand Down

0 comments on commit 012081a

Please sign in to comment.