DQN的Tensorboard-强化学习

时间:2019-07-27 17:37:51

标签: tensorflow keras tensorboard

我正在运行强化学习培训脚本;因此,该模型每集训练一次:

tensorboard = TensorBoard(log_dir= 'thepathIvespecified')      

model = Sequential()
model.add(InputLayer(batch_input_shape=(1, input_dimension)))

model.add(Dense(32))
model.add(Activation('relu'))

model.add(Dense(parameters["max_products"]))
model.add(Activation('linear'))

model.compile(loss='mse', optimizer='adam', metrics=['mae'])  

history = model.fit(current_state_input,
                                target_vec.reshape(-1, parameters["max_products"]), 
                                epochs = 1, 
                                verbose = 0,
                                callbacks=[tensorboard])

但是,它在文件夹的每一步都会创建一个文件。

如何正确显示Tensorboard?

0 个答案:

没有答案