Articles in the category of pytorch实战

  1. Home
  2. pytorch实战

pytorch 逐层加载模型参数

pytorch 逐层加载模型参数1.现象描述当对模型结构的某些层做了微小调整之后，导致该层参数的shape发生了微小变化导致无法对整个模型进行加载，这时可以考虑通过逐层加载的方式来跳过某些层完成对保存好的模型的加载。2.代码实现# 模型参数逐层加载 ckpt_state_dict = torch.load("ckpt/best_att_wer_0.01526717557251908...

  • jupiter
  • January 8, 2022
  • No comments

    • Pytorch 实战：使用Resnet18实现对是否戴口罩进行图片分类

    1.实验环境torch = 1.6.0torchvision = 0.7.0matplotlib = 3.3.3 # 绘图用progressbar = 2...

  • jupiter
  • November 25, 2021
  • No comments
    •

    Pytorch 实战：卷积神经网络(CNN)实现MINIST手写数字识别

    实验环境torch = 1.6.0torchvision = 0.7.0matplotlib = 3.3.3 # 绘图用progressbar = 2.5...

  • jupiter
  • January 15, 2021
  • 2 comments
    •

    Pytorch 实战：BP神经网络实现MINIST实现手写数字识别(单层感知机)

    Pytorch 实战：BP神经网络实现MINIST实现手写数字识别(单层感知机)实验环境torch = 1.6.0torchvision = 0.7.0m...

  • jupiter
  • January 15, 2021
  • No comments