2020年10月30日

torch.nn.MSELoss()函数解读

摘要: 函数作用torch.nn.MSELoss() 求predict和target之间的loss。 代码示例单个求其loss: crit = nn.MSELoss() # target = torch.Tensor(1) # target[0] = 10 # res = torch.Tensor(1) # 阅读全文

posted @ 2020-10-30 15:22 cltt 阅读(8018) 评论(0) 推荐(0) 编辑

torch.cuda.is_available() False

摘要: 这里的cuda=10.1 需要: pip install --default-timeout=1000 torch==1.3.1 torchvision==0.4.2 -i https://pypi.tuna.tsinghua.edu.cn/simple torch ,torchvision 要和c 阅读全文

posted @ 2020-10-30 09:18 cltt 阅读(443) 评论(0) 推荐(0) 编辑

导航