下载直接使用即可,可以自己增加权重保存 、 path = 'weight/cifar_resnet.pth' torch.save(model.state_dict(),path)#加到循环里面去 epoch可以设置为10提高训练准确率。
-
Notifications
You must be signed in to change notification settings - Fork 0
thewzm/resnet_class_cifar10
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published