pytorch大体来说分为三个部分,第一个部分是数据准备,第二个部分是model搭建,第三个部分是训练。
pytorch的数据准备,大致为:data-->tensor data-->data set(data target)-->data load(加batch)
pytorch的模型搭建:继承自torch.nn.module,一定要重写一个forward的函数,将这个模型定义为class Net
训练:定义一个Net,然后加入学习率等参数,选择合适的优化算法和loss函数
pytorch大体来说分为三个部分,第一个部分是数据准备,第二个部分是model搭建,第三个部分是训练。
pytorch的数据准备,大致为:data-->tensor data-->data set(data target)-->data load(加batch)
pytorch的模型搭建:继承自torch.nn.module,一定要重写一个forward的函数,将这个模型定义为class Net
训练:定义一个Net,然后加入学习率等参数,选择合适的优化算法和loss函数