pytorch-YOLOv4 icon indicating copy to clipboard operation
pytorch-YOLOv4 copied to clipboard

Published 20 hours ago • verified publisher icon Tianxiaomo

[BUG]model.moduel,state_dict()

Open zkailinzhang opened this issue 3 years ago • 0 comments

https://github.com/Tianxiaomo/pytorch-YOLOv4/blob/a65d219f9066bae4e12003bd7cdc04531860c672/train.py#L451 to: torch.save(model.module.state_dict(), save_path)

zkailinzhang avatar Jan 30 '22 02:01 zkailinzhang