我出错的原因是:
利用nn.MSELoss()计算时,输入给的是eagerTensor格式,要求是torch里的tensor格式。
首先我利用了keras调用的vgg预训练模型。
vgg的输入是numpy,输出是eagerTensor格式(与pytorch的tensor格式不一致),所以需要将eagerTensor转为numpy再torch.tensor转为pytorch接受的格式。
我出错的原因是:
利用nn.MSELoss()计算时,输入给的是eagerTensor格式,要求是torch里的tensor格式。
首先我利用了keras调用的vgg预训练模型。
vgg的输入是numpy,输出是eagerTensor格式(与pytorch的tensor格式不一致),所以需要将eagerTensor转为numpy再torch.tensor转为pytorch接受的格式。