deep learning
weixin_40277254
这个作者很懒,什么都没留下…
展开
-
深度学习 batch_size
1. batch_size、iteration、epoch简介 batch_size:每批数据量的大小。深度学习通常用SGD的优化算法进行训练,也就是一次(1 个iteration)一起训练batch_size个样本,计算它们的平均损失函数值,来更新参数。 iteration:1个iteration即迭代一次,也就是用batchsize个样本训练一次。 epoch:1个epoch指用训练集中...原创 2019-01-04 16:19:39 · 605 阅读 · 0 评论 -
终端训练模型指令
在终端执行程序时指定GPU CUDA_VISIBLE_DEVICES=0 python your_file.py # 指定GPU集群中第一块GPU使用,其他的屏蔽掉 在python代码中指定GPU import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块gpu 终端指定GPU,后台挂起 CUDA_VISIBLE_DE...原创 2019-04-18 20:50:11 · 789 阅读 · 0 评论