【小土堆】Pytorch学习笔记_P27/28/29_完整的模型训练套路

(0)摘要

# 课程链接

PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】_哔哩哔哩_bilibili

# 笔记内容

(1)P27_完整模型训练的套路(一)

(2)P28_完整模型训练的套路(二)

(3)P29_完整模型训练的套路(三)


(1)完整模型训练的套路(一)

# (1)完整网络训练的搭建。

                1)规范化。创建 models.py 用于存放网络模型,这是一种规范。

# -*- coding=utf-8 -*- 
# github: night_walkiner
# csdn: 潘迪仔

import torch
import torch.nn as nn


# 创建网络模型
class Models(nn.Module):

    def __init__(self):
        super(Models, self).__init__()
        self.model = nn.Sequential(
            # (1)conv1、maxpool

你可能感兴趣的:(pytorch学习笔记_小土堆,pytorch)