Pytorch
Sunny_Jie
悄点儿,我思慕思慕。
展开
-
torch模型的保存与加载
两种方式''' 第一种方式 模型整体保存,占用空间会比较大'''torch.save(net, "../model/model.pkl")torch.load("")''' 第二种方式 保存模型参数,在加载模型参数之前,必须先建立模型'''torch.save(net.state_dict(), "params.pkl")net.load_state_dict()下面是整个流程的代码import torch# data 数据加载import nu原创 2020-08-29 08:39:17 · 1631 阅读 · 0 评论 -
Object Detection finetuning tutorial + Pytorch代码实战练兵
本篇文章的学习资料来源于Pytorch官网教程内容,旨在加深对Pytorch的整体使用把握和代码细节理解,如有误读之处,请提出来,笔者愿向大家学习。声明:英文说明我就不翻译了,直接搬过来,因为很容易看懂。正文:For this tutorial, we will be finetuning a pre-trained Mask R-CNN model in the Penn-Fudan Da...原创 2020-05-03 23:26:30 · 1166 阅读 · 0 评论 -
PyTorch里如何利用TensorBoard--详解
在这份文档中,将记录以下几点:读取数据,并作适当的数据转换;设置TensorBoard;写入TensorBoard相关内容;利用TensorBoard查看模型结构;利用TensorBoard创建可视化的交互界面;特别是在第5点中,我们将看到:查看训练数据的几种方式;在训练时如何追踪模型的性能;训练结束后,如何评估模型的性能。本文所用数据集为 CIFAR-10。# im...原创 2020-04-20 18:01:54 · 12049 阅读 · 1 评论