NJU1healer的个人博客分享 http://blog.sciencenet.cn/u/NJU1healer

博文

按标题搜索
[转载]DataLoader完整的参数
2020-10-5 14:56
DataLoader完整的参数表如下: DataLoader完整的参数表如下: classtorch.utils.data.DataLoader( dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, & ...
个人分类: Pytorch|4488 次阅读|没有评论
Pytorch批量对一化,梯度消失和爆炸
2020-10-5 13:57
1.nn.BatchNorm1d(num_features) 1.对小批量(mini-batch)的2d或3d输入进行批标准化(Batch Normalization)操作 2.num_features: 来自期望输入的特征数,该期望输入的大小为'batch_size x num_features ' &n ...
个人分类: Pytorch|2459 次阅读|没有评论
正则表达式re
2020-9-30 16:44
参考博客: Python3的re.match()、re.search()和re.findall() Python:正则表达式(一):search()、match()、findall() 的区别 Python:正则表达式(二):如何使用re.search()返回的匹配对象中的具体内容呢?? 正则就是提取一段特征数据,用(.*?)代替。 总结的特点区别: (.*) 贪婪匹配 ...
个人分类: Python|2553 次阅读|没有评论
Pytorch to(device)多GPU并行和查看GPU信息
2020-9-30 15:34
(一)参考博客: Pytorch to(device) Pytorch:多GPU训练网络与单GPU训练网络保存模型的区别 (二)常用的查看GPU信息的命令 为什么将数据转移至GPU的方法叫做.cuda而不是.gpu,就像将数据转移至CPU调用的方法是.cpu?这是因为GPU的编程接口采用CUDA,而目前并不是所有的GPU都支持C ...
个人分类: Pytorch|4348 次阅读|没有评论
model.train()与model.eval()的用法
2020-9-25 10:31
1.问题来源 看别人的面经时,浏览到一题,问的就是这个。自己刚接触pytorch时套用别人的框架,会在训练开始之前写上model.trian(),在测试时写上model.eval()。然后自己写的时候也就保留了这个习惯,没有去想其中原因。 2.用法(区别) 在经过一番查阅之后,总结如下: &nbs ...
个人分类: Pytorch|6161 次阅读|没有评论

本页有 5 篇博文因作者的隐私设置或未通过审核而隐藏

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2023-2-1 13:35

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部