- 博客(2)
- 问答 (1)
- 收藏
- 关注
原创 Pytorch中.backward()用法记录
backward()使用举例#定义变量import torchdata_list = [1,2,3,4,5]data_in = torch.Tensor(data_list)data_1 = data_in * 2data_2 = data_1 * 3data_3 = data_2 * 4#错误方式data_3.backward()报错:#正确方式1data_3.backward(torch.ones_like(data_3))print(data_in.grad)#方式
2022-05-20 16:38:31
307
原创 2021-07-08
错误:RuntimeError: expected scalar type Byte but found Float错误:RuntimeError: expected scalar type Byte but found Float问题描述:在使用pytorch的nn.Conv2d()编写卷积神经网络时,提示报错。网络结构class CNN(nn.Module): def __init__(self): super().__init__() self.co
2021-07-08 23:24:35
470
3
空空如也
在windows运行pytorch环境下的程序报错
2022-04-18
TA创建的收藏夹 TA关注的收藏夹
TA关注的人