随着时间的推移,TFLearn训练越来越慢

时间:2017-11-18 10:41:43

标签: python tensorflow tflearn

我正在使用this CNN模型训练我的网络。 首先我在我的train_model.py中初始化它:

 model = cnn_model.googlenet(WIDTH, HEIGHT, 3, LR, output=8, model_name=MODEL_NAME)

然后我定义了训练循环,在其中,我放置了 model.fit 函数:

model.fit({'input': batch_x}, {'targets': batch_y}, n_epoch=1, snapshot_step=2510, show_metric=True, run_id=MODEL_NAME)

由于某种原因,网络随着时间的推移而减慢,并且在大约13小时之后,tf抛出异常:内存错误

0 个答案:

没有答案
相关问题