摘要: 每次进行数据训练时,如果数据集过多不适合把整个数据打包进行训练,那么这种时候就需要把训练数据分批进行训练。 设置数据: x = torch.linspace(1, 10, 10) # x data (torch tensor) y = torch.linspace(10, 1, 10) # y da 阅读全文
posted @ 2021-11-26 14:32 Sunshine_y 阅读(81) 评论(0) 推荐(0) 编辑
摘要: 一般来说,神经网络是用来训练数据,做实验等等,但每次关闭神经网络或者关机后,训练好的结果等都自动清除,为了解决这一问题,需要对神经网络训练后的结果进行保存、提取。 一、保存神经网络 搭建神经网络、训练数据、优化等操作都需要在save()函数中进行,代码如下: def save(): #快速搭建 ne 阅读全文
posted @ 2021-11-26 13:44 Sunshine_y 阅读(127) 评论(0) 推荐(0) 编辑
摘要: 快速搭建 无论是搭建回归类型的神经网络也好,分类类型的也罢,都是通过 class Net(torch.nn.Module) 搭建,本文介绍另一种搭建方法,这种方法可以更快速的搭建net 。 #快速搭建 net = torch.nn.Sequential( torch.nn.Linear(2, 10) 阅读全文
posted @ 2021-11-26 13:27 Sunshine_y 阅读(23) 评论(0) 推荐(0) 编辑