我正在做一个使用 CNN 模型对 CT 扫描图像进行分类的项目,图像大小很大,我想使用批处理的想法将其输入网络,尝试使用以下代码:
# train_data size = 5460
num_epochs = 14
batch_size = 390
batch = 0
print("Starting training...")
for epoch in range(num_epochs):
train_batch = train_data[batch:batch_size]
batch += batch_size
batch_size += batch_size
ep_loss = 0
for data in train_batch:
X = data[0]
Y = data[1]
_, c = sess.run([optimizer, cost], feed_dict={x_img: X, y_label: Y})
我的问题是:
1-这是进行批处理的正确方法吗?还是有更好的方法?
2-使用上面的代码,我正在使用“AdamOptimizer”,对于批处理的想法,这是一种很好的优化技术,还是我应该使用另一种?