Testing error

[xxktju]

I build a new model, after quantization aware training this model, we meet the following test error.

can u give me some suggestions?

[colorjam]

@xxktju Hi, it seems the shape of max_val in the saved checkpoint is inconsistent with the model state_dict.
You can try to preprocess the saved checkpoint like:

ckpt = torch.load(ckpt_path)['state_dict']
for k, v in ckpt.items():
    if 'max_val' in k:
        ckpt[k] = v.unsqueeze(0)
model.load_state_dict(ckpt)

[MonicaHui]

@xxktju Hi, it seems the shape of max_val in the saved checkpoint is inconsistent with the model state_dict.
You can try to preprocess the saved checkpoint like:

ckpt = torch.load(ckpt_path)['state_dict']
for k, v in ckpt.items():
    if 'max_val' in k:
        ckpt[k] = v.unsqueeze(0)
model.load(ckpt)

Hi, thanks for your suggestion, I recently concurred with this problem and solved by following your guide. 👍
However, the last sentence should be "model.load_state_dict(ckpt)" I think.
Thank you once again!