该代码编写过程中主要有3个雷:
1.输入数据的类型开头已经被dtype=float.32确定了,要看数据集符不符合标准
2.使用新数据集要注意x的维度
3.使用pytorch的dataset,dataloader要注意其与win系统不兼容的问题,训练循环要在一个if语句里,是这样:
if __name__ ="__main__":
for epoch in range(100):
for i, data in enumerate(trainloader(),0):
······
2023-11-14 17:18:58
该代码编写过程中主要有3个雷:
1.输入数据的类型开头已经被dtype=float.32确定了,要看数据集符不符合标准
2.使用新数据集要注意x的维度
3.使用pytorch的dataset,dataloader要注意其与win系统不兼容的问题,训练循环要在一个if语句里,是这样:
if __name__ ="__main__":
for epoch in range(100):
for i, data in enumerate(trainloader(),0):
······