摘要:
问题说明:即使在train表明了model.to(device),但仍然会报错 input是torch.cuda.tensor.float,但weight 是torch.tensor.float即网络的权重不是cuda类型 解决:发现在model中的forward中,也定义了一个卷积核,其余卷积核皆 阅读全文
posted @ 2022-06-16 20:56
原来是只呆燕
阅读(206)
评论(0)
推荐(0)
摘要:
点击查看代码 """ 可视化结果 """ import matplotlib.pyplot as plt f=open('/home/yanhua/Desktop/ncc-600_0.0004.txt','r') data=f.read().splitlines() f.close() test=d 阅读全文
posted @ 2022-06-08 10:51
原来是只呆燕
阅读(35)
评论(0)
推荐(0)