摘要： 一、入门 二、计算图的概念 三、张量的概念 四、张量的类型 五、操作 阅读全文

摘要： 一、main.py import torch import tqdm from sklearn.metrics import roc_auc_score from torch.utils.data import DataLoader import os import numpy as np from 阅读全文

摘要： pytorch保存模型非常简单，主要有两种方法： 只保存参数；（官方推荐） 保存整个模型 (结构+参数)。 由于保存整个模型将耗费大量的存储，故官方推荐只保存参数，然后在建好模型的基础上加载。本文介绍两种方法，但只就第一种方法进行举例详解。 一、只保存参数 1.保存 一般地，采用一条语句即可保存参数 阅读全文