search

GOATmessi8 / ASFF

yolov3 with mobilenet v2 and ASFF
GNU General Public License v3.0
1.05k stars 216 forks source link

AttributeError: 'YOLOv3' object has no attribute 'module'！ #80

Open xaohuicai opened 4 years ago

xaohuicai commented 4 years ago

model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i]) torch.save(model.module.state_dict..) all meet this problem,I don't know how to solve this.

missbook520 commented 4 years ago

model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i]) torch.save(model.module.state_dict..) all meet this problem,I don't know how to solve this.

may be you can replace it with model.state_dict..

missbook520 commented 4 years ago

model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i]) torch.save(model.module.state_dict..) all meet this problem,I don't know how to solve this.

are you meet the error? RuntimeError: Function 'MulBackward0' returned nan values in its 1th output.

xaohuicai commented 4 years ago

model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i]) torch.save(model.module.state_dict..) all meet this problem,I don't know how to solve this.

may be you can replace it with model.state_dict.. thanks!I will try it.