MLS2021

加载断点path

5

resnet.load_state_dict((torch.load("/root/data/others/yaoganbisai/code_6_7/models/pretrained_model/resnet50-19c8e357.pth"))) if args.model == 'deeplabv3_version_1'中，设置了path之后，会报错。错误在上面。这个resnet里面的pth是怎么生成的。

Unknown CUDA arch or GPU not supported

1

First of all, thank you for your work and your results. It's great. I encountered this problem when installing the environment: > raise ValueError("Unknown CUDA arch ({}) or GPU not...

问题请教

6

您好，作为变化检测的初学者，在阅读和运行了您的代码之后，有个问题需要向您请教： if return_aux: pred_v = self.conv_out_v(feats_v[-1]) pred_v = F.interpolate(pred_v, size=pred.shape[2:]) return pred, pred_v else: return pred 如果 return_aux = True ，返回的就是一个turple。我直接开始训练的话就会报错： if not (target.size() == input.size()): AttributeError: 'tuple' object has...