We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 67a4094 commit b64126bCopy full SHA for b64126b
Tests/TestCCTPretrained.py
@@ -295,7 +295,7 @@ def deepQuantTestCCT():
295
296
# FBRANCASI: Load original floating point model
297
originalModel = cct_2_3x2_32()
298
- checkpointPath = "/Users/federicobrancasi/Documents/DeepQuant/Tests/Data/checkpoint_epoch_200_cct2_cifar10.pth"
+ checkpointPath = "./Tests/Data/checkpoint_epoch_200_cct2_cifar10.pth"
299
checkpoint = torch.load(checkpointPath, map_location="cpu")
300
originalModel.load_state_dict(checkpoint["model_state_dict"])
301
originalModel = originalModel.eval().to(device)
0 commit comments