我正在使用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抛出异常:内存错误