赞
踩
https://www.cnblogs.com/xiaodai0/p/10413711.html
这个博客建议个格式
torch.load("0.9472_0048.weights",map_location='cpu')
下面格式成功运行
torch.load(xxx, map_location=torch.device('cpu')) # 这是终端报错提示我使用的格式
不管读的什么模型都加上,就没问题了
torch.load(self.config.pre_trained, map_location=torch.device('cpu'))
1 solver.py 这里很奇怪,默认是True虽然我 --cuda False了
2 parser.add_argument('--cuda', type=bool, default=False) 这里也默认是False了
- def build_model(self):
-
- self.net_bone = build_model(base_model_cfg) # model.build_model, base_model_cfg = 'resnet'
-
- print('=>1 self.config.cuda:', self.config.cuda) # todo =>cuda?//
- # self.config.cuda = False
- print('=>11 self.config.cuda:', self.config.cuda) # todo =>cuda?//
- '''if self.config.cuda:
- self.net_bone = self.net_bone.cuda()'''
- self.net_bone.eval() # use_global_stats = True # todo =>?//
- self.net_bone.apply(weights_init)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。