- 博客(5)
- 收藏
- 关注
RSS订阅原创 pytorch多卡并行模型的保存与载入
多卡并行模型的保存与载入当模型是在数据并行方式在多卡上进行训练的训练和保存，那么载入的时候也是一样需要是多卡。并且，load_state_dict()函数的调用要放在DataParallel()之后，而model.cuda()苏在的位置无影响。model = DefinedNetwork()model = torch.nn.parallel.DataParallel(model, devic...
2019-11-29 11:03:48
3399
2
原创 Java基本语法坑
java中数组的长度访问有属性length，而字符串的长度访问用方法length()java中List里面的size() 返回此列表中的元素数 .Map里面的size() 返回此映射中的键-值映射关系数。set里面的size() 返回 set 中的元素数（其容量）。java中当你写了一个构造器（构造函数），那么编译器就不会为你创建默认构造器（构造函数）。java中构造器可以调...
2018-12-18 14:41:56
178
原创 scikit-learn学习笔记
在scikit-learn中，LinearRegression,Lasso,Ridge,ElasticNet都是回归模型。而LogisticRegression是分类模型。Lasso是LinearRegression带有L1正则项，Ridge是LinearRegression有L2正则项，ElasticNet是LinearRegression同时有L1和L2正则项。而LogisticRe...
2018-12-18 14:35:54
401
原创 python获取一个目录下所有jpg文件
输入：文件夹目录名称输出：以列表形式返回该目录下包括其子目录下所有的后缀名为.jpg的文件名import osdef get_fileNames(rootdir): fs = [] for root, dirs, files in os.walk(rootdir,topdown = True): for name in files: _, ending = ...
2018-12-18 14:22:11
5857
2
原创 ubuntu下文件和文件夹复制
pipy国内镜像目前有：http://pypi.douban.com/ 豆瓣http://pypi.hustunique.com/ 华中理工大学http://pypi.sdutlinux.org/ 山东理工大学http://pypi.mirrors.ustc.edu.cn/ 中国科学技术大学对于pip这种在线安装的方式来说，很方便，但网络不稳定的话很要命。使用国内镜像相对好一些，如果...
2018-12-18 14:11:30
4418
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人