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

Published 20 hours ago • verified publisher icon bubbliiiing

你好

Open wujinxiao opened this issue 3 years ago • 3 comments
trafficstars

你好 我想问一下 我将Resnet换成了SENet 请问如果预训练权重不可使用了 那我该怎么做才能训练

wujinxiao avatar Dec 08 '21 13:12 wujinxiao

应该直接有se-resnet那种库吧，都提供权值的

bubbliiiing avatar Dec 08 '21 14:12 bubbliiiing

你好 我找到了se-resnet的库： import torch.hub hub_model = torch.hub.load( 'moskomule/senet.pytorch', 'se_resnet50', pretrained=True,) 请问需要把这个加在哪里

wujinxiao avatar Dec 09 '21 02:12 wujinxiao

那你就根据现在有的这个主干修改一下呗，载入权值和网络加载的部分不都有吗

bubbliiiing avatar Dec 12 '21 16:12 bubbliiiing