![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
心心喵
这个作者很懒,什么都没留下…
展开
-
[pytorch] pytorch_model.bin 和 training_args.bin 的区别
在使用 PyTorch 进行深度学习训练时,经过训练的模型会被保存为这个文件,其中包含了模型的权重参数。文件保存了这些训练参数,包括学习率、优化器类型、训练批次大小、训练时的超参数等等。当需要重新加载模型进行训练或进行进一步的微调时,这个文件可以用来还原训练参数的配置。: 这是保存了训练参数的二进制文件。在使用 PyTorch 进行训练时,可以使用一个包含了各种训练参数的对象(通常是一个命名元组或类实例)来配置训练过程。则包含了训练模型时使用的各种训练参数的配置信息。包含了经过训练的模型的权重参数,而。原创 2023-05-11 11:35:21 · 2594 阅读 · 0 评论 -
[pytorch] torch.clamp
将输入中的所有元素限制在[min, max]范围内。原创 2023-03-22 15:57:55 · 95 阅读 · 0 评论 -
[pytorch] pytorch 优雅画图
tensorboard太丑了,用plt。原创 2022-10-08 14:18:31 · 565 阅读 · 1 评论 -
[pytorch] torch.cuda.FloatTensor报错
参考:RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [4, 512, 16, 16]], which is output 0 of ConstantPadNdBackward, is at version 1; expected version 0 instead. Hint: enabl原创 2021-10-28 13:37:26 · 2683 阅读 · 1 评论 -
[pytorch] PyTorch—torchvision.models导入预训练模型—残差网络代码讲解
参考:https://blog.csdn.net/wsp_1138886114/article/details/83787181PyTorch框架中torchvision模块下有:torchvision.datasets、torchvision.models、torchvision.transforms这3个子包。关于详情请参考官网: http://pytorch.org/docs/master/torchvision/index.html。具体代码可以参考github: https://github原创 2021-07-06 23:15:52 · 758 阅读 · 1 评论 -
[pytorch] 服务器上使用tensorboardX可视化
参考: https://github.com/lanpa/tensorboardX一、关于tensorboardXtensorflow框架可以使用tensorboard进行可视化。pytorch环境下,下载tensorflow也可以使用tensorboardX为可视化工具。TensorboardX支持scalar, image, figure, histogram, audio, text,...原创 2020-03-12 19:39:46 · 2184 阅读 · 0 评论 -
[pytorch] pytorch安装和更新
1、使用Pip更新Pytorch和torchvision列举pip当前可以更新的所有安装包pip list --outdated --format=legacy更新pytorch和torchvision安装包pip install --upgrade pytorch torchvision2、使用conda更新Pytorch和torchvision建议将其添加soumith为您的Ana...原创 2020-03-05 22:11:43 · 2865 阅读 · 0 评论