2020年10月19日
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
摘要： Pytorch使用过程中 torchsummary.summary( )中出现了上述错误，torchsummary是应用在pytorch中的一种结构表达方式。 # 1 if __name__ == '__main__': model = fishnet99() torchsummary.summar 阅读全文
posted @ 2020-10-19 20:03 blog_wu 阅读(974) 评论(0) 推荐(0) 编辑