csdn421的博客

原创 pytorch

pytorch大体来说分为三个部分，第一个部分是数据准备，第二个部分是model搭建，第三个部分是训练。pytorch的数据准备，大致为：data-->tensor data-->data set(data target)-->data load(加batch)pytorch的模型搭建：继承自torch.nn.module，一定要重写一个forward的函数，将这个模型定义为class N