hello, cheng,
I want to quantization of Unet model, and searched your blog in csdn, https://blog.csdn.net/lai_cheng/article/details/118961420, one question need to confirm with you, that is as the title "where is to download the dataset of coco_pose_iter_440000.pth.tar and put it in the model folder" in step 2?
-
how to download the dataset of coco_pose_iter_440000.pth.tar, is it just for the VGG model?
If I want to quant my unet model, could I use my checkpoint.pth for my unet model?
-
the model foler need to create by myself? because I could not find it when git clone the your code?
-
when I run pth_to_int.py, found the issue as below:root@sse-lg-112-129:~/mn_ws/pytorch_model_quantization# python3 pth_to_int.py
Traceback (most recent call last):
File "pth_to_int.py", line 42, in
state_dict = torch.load('./models/checkpoint_epoch1_unet.pth')['state_dict']
KeyError: 'state_dict'
Does any model file pth including the key“state_dict”? because I could cat this pth file, need your help, thanks very much!
hello, cheng,
I want to quantization of Unet model, and searched your blog in csdn, https://blog.csdn.net/lai_cheng/article/details/118961420, one question need to confirm with you, that is as the title "where is to download the dataset of coco_pose_iter_440000.pth.tar and put it in the model folder" in step 2?
how to download the dataset of coco_pose_iter_440000.pth.tar, is it just for the VGG model?
If I want to quant my unet model, could I use my checkpoint.pth for my unet model?
the model foler need to create by myself? because I could not find it when git clone the your code?
when I run pth_to_int.py, found the issue as below:root@sse-lg-112-129:~/mn_ws/pytorch_model_quantization# python3 pth_to_int.py
Traceback (most recent call last):
File "pth_to_int.py", line 42, in
state_dict = torch.load('./models/checkpoint_epoch1_unet.pth')['state_dict']
KeyError: 'state_dict'
Does any model file pth including the key“state_dict”? because I could cat this pth file, need your help, thanks very much!