- 博客(4)
- 资源 (8)
- 收藏
- 关注
原创 PyTorch 模型的保存和加载
1. Pytorch 加载完整模型的参数保存加载整个模型 # 保存整个模型torch.save (model_object, 'model.pk1')# 加载整个模型 model = torch.load('model.pkl')保存模型的参数 (推荐使用)# 模型参数保存torch.save (model_object.state_dict(), 'params....
2018-08-18 19:03:39 4706 1
原创 深度学习中的数据增强
关于计算机视觉领域数据增强的一些常用的方法为什么需要数据增强 :一般而言,比较成功的神经网络需要大量的参数,许许多多的神经网路的参数都是数以百万计,而使得这些参数可以正确工作则需要大量的数据进行训练,而实际情况中数据并没有我们想象中的那么多数据增强的作用 :增加训练的数据量,提高模型的泛化能力增加噪声数据,提升模型的鲁棒性如何获得大量的数据 :一种方法是获得新的数...
2018-08-18 17:06:42 60090 28
原创 Python 之 os
Python 之 os :最近在看的一些代码中许多都用到了 Python 中的 os 模块,于是决定整理一下这个模块的一些常见的用法 :os.getcwd() 返回当前目录的路径os.listdir () 返回当前目录下的所有文件的名称 以列表的形式返回 os.path 系列 : os.path.abspath(path) : 返回绝对路径 os.path.split...
2018-08-08 21:20:28 249
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人