会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
买白菜不用券
博客园
首页
新随笔
联系
管理
订阅
2019年1月14日
记录:测试本机下使用 GPU 训练时不会导致内存溢出的最大参数数目
摘要: 本机使用的 GPU 是 GeForce 840M,2G 显存,本机内存 8G。 试验时,使用 vgg 网络,调整 vgg 网络中的参数,使得使用对应的 batch_size 时不会提示内存溢出。使用的是 mnist 数据集,图片尺寸是28×28。结果如下: # batch_size = 1 # 参数
阅读全文
posted @ 2019-01-14 17:12 买白菜不用券
阅读(698)
评论(0)
推荐(0)
编辑
公告