{"action":"created","issue":{"url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose/issues/145","repository_url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose","labels_url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose/issues/145/labels{/name}","comments_url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose/issues/145/comments","events_url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose/issues/145/events","html_url":"https://github.com/MVIG-SJTU/AlphaPose/issues/145","id":368013513,"node_id":"MDU6SXNzdWUzNjgwMTM1MTM=","number":145,"title":"用重新训练得到的模型进行检测时权重维度报错","user":{"login":"ZhongXiaoFang","id":40446895,"node_id":"MDQ6VXNlcjQwNDQ2ODk1","avatar_url":"https://avatars1.githubusercontent.com/u/40446895?v=4","gravatar_id":"","url":"https://api.github.com/users/ZhongXiaoFang","html_url":"https://github.com/ZhongXiaoFang","followers_url":"https://api.github.com/users/ZhongXiaoFang/followers","following_url":"https://api.github.com/users/ZhongXiaoFang/following{/other_user}","gists_url":"https://api.github.com/users/ZhongXiaoFang/gists{/gist_id}","starred_url":"https://api.github.com/users/ZhongXiaoFang/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/ZhongXiaoFang/subscriptions","organizations_url":"https://api.github.com/users/ZhongXiaoFang/orgs","repos_url":"https://api.github.com/users/ZhongXiaoFang/repos","events_url":"https://api.github.com/users/ZhongXiaoFang/events{/privacy}","received_events_url":"https://api.github.com/users/ZhongXiaoFang/received_events","type":"User","site_admin":false},"labels":[],"state":"open","locked":false,"assignee":null,"assignees":[],"milestone":null,"comments":2,"created_at":"2018-10-09T02:08:43Z","updated_at":"2018-10-10T10:16:52Z","closed_at":null,"author_association":"NONE","body":"您好，大神，根据您之前的指导初步完成了在COCO数据上的训练，当用模型进行检测的时候却出现了权重维度大小不匹配的情况，折腾很久没有解决，求指点。。。谢谢\r\nLoading pose model from ./models/sppe/model_49.pkl\r\nTraceback (most recent call last):\r\n  File \"demo.py\", line 56, in <module>\r\n    pose_model = InferenNet_fast(4 * 1 + 1, pose_dataset)\r\n  File \"/home/zhong/AlphaPose/SPPE/src/main_fast_inference.py\", line 61, in __init__\r\n    model.load_state_dict(torch.load('./models/sppe/model_49.pkl'))\r\n  File \"/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py\", line 719, in load_state_dict\r\n    self.__class__.__name__, \"\\n\\t\".join(error_msgs)))\r\nRuntimeError: Error(s) in loading state_dict for FastPose:\r\n\tsize mismatch for conv_out.bias: copying a param of torch.Size([33]) from checkpoint, where the shape is torch.Size([17]) in current model.\r\n\tsize mismatch for conv_out.weight: copying a param of torch.Size([33, 128, 3, 3]) from checkpoint, where the shape is torch.Size([17, 128, 3, 3]) in current model."},"comment":{"url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose/issues/comments/428517907","html_url":"https://github.com/MVIG-SJTU/AlphaPose/issues/145#issuecomment-428517907","issue_url":"https://api.github.com/repos/MVIG-SJTU/AlphaPose/issues/145","id":428517907,"node_id":"MDEyOklzc3VlQ29tbWVudDQyODUxNzkwNw==","user":{"login":"Jeff-sjtu","id":15179660,"node_id":"MDQ6VXNlcjE1MTc5NjYw","avatar_url":"https://avatars2.githubusercontent.com/u/15179660?v=4","gravatar_id":"","url":"https://api.github.com/users/Jeff-sjtu","html_url":"https://github.com/Jeff-sjtu","followers_url":"https://api.github.com/users/Jeff-sjtu/followers","following_url":"https://api.github.com/users/Jeff-sjtu/following{/other_user}","gists_url":"https://api.github.com/users/Jeff-sjtu/gists{/gist_id}","starred_url":"https://api.github.com/users/Jeff-sjtu/starred{/owner}{/repo}","subscriptions_url":"https://api.github.com/users/Jeff-sjtu/subscriptions","organizations_url":"https://api.github.com/users/Jeff-sjtu/orgs","repos_url":"https://api.github.com/users/Jeff-sjtu/repos","events_url":"https://api.github.com/users/Jeff-sjtu/events{/privacy}","received_events_url":"https://api.github.com/users/Jeff-sjtu/received_events","type":"User","site_admin":false},"created_at":"2018-10-10T10:16:52Z","updated_at":"2018-10-10T10:16:52Z","author_association":"COLLABORATOR","body":"你好，\r\n关于问题1&4：33的heatmap是包括了coco和mpii数据集的输出，在inference的demo中，取了前17个heatmap输出作为coco数据集的结果，如果取后16个heatmap即是mpii数据集的关键点。\r\n问题2：batchsize需要根据你使用的gpu数量以及显存大小决定。\r\n问题3：预训练模式是指训练到一定程度后，可以load模型，加入dpg模式训练继续在原本模型的基础上训练。如果想对我们放出来的模型在coco数据集上自行训练，需要自行对最后一层卷积层复制参数。\r\n"}}
{"actor":{"display_login":"Jeff-sjtu"}}