pytorch: pokemon-resnet-代码数据集

阅读量：

pokemon数据集详情请参见：缦旋律的资源合集

文章结构概述

一. 创建一个用于获取图像及其对应标签的Pokemon类
- 二. 构建自定义的Resnet模型
- - 1. 设计resblock模块：
  - 1. 组装完整的Resnet结构：
- 三. 配置若干关键超参数
- 四. 导入训练所需的数据集
- 五. 初始化网络模型，并设定损失函数、优化器及评估指标
- 六. 启动训练流程，同时执行验证操作

复制代码

    import torch
    from torch import nn
    from torch.nn import functional as F
    from torchvision import transforms
    from torch.utils.data import DataLoader,Dataset
    from torch import optim
    import os
    import csv
    from PIL import Image
    import warnings
    warnings.simplefilter('ignore')

全部评论 (0)

还没有任何评论哟~