你好
wujinxiao opened this issue · comments
你好 我想问一下 我将Resnet换成了SENet 请问如果预训练权重不可使用了 那我该怎么做才能训练
应该直接有se-resnet那种库吧,都提供权值的
你好 我找到了se-resnet的库:
import torch.hub
hub_model = torch.hub.load(
'moskomule/senet.pytorch',
'se_resnet50',
pretrained=True,)
请问需要把这个加在哪里
那你就根据现在有的这个主干修改一下呗,载入权值和网络加载的部分不都有吗