机器学习
新技能get
每天进步一点点
展开
-
pytorch多卡并行模型的保存与载入
多卡并行模型的保存与载入 当模型是在数据并行方式在多卡上进行训练的训练和保存,那么载入的时候也是一样需要是多卡。并且,load_state_dict()函数的调用要放在DataParallel()之后,而model.cuda()苏在的位置无影响。 model = DefinedNetwork() model = torch.nn.parallel.DataParallel(model, devic...原创 2019-11-29 11:03:48 · 3406 阅读 · 2 评论 -
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 · 404 阅读 · 0 评论