摘要: 本机使用的 GPU 是 GeForce 840M,2G 显存,本机内存 8G。 试验时,使用 vgg 网络,调整 vgg 网络中的参数,使得使用对应的 batch_size 时不会提示内存溢出。使用的是 mnist 数据集,图片尺寸是28×28。结果如下: # batch_size = 1 # 参数 阅读全文
posted @ 2019-01-14 17:12 买白菜不用券 阅读(694) 评论(0) 推荐(0) 编辑