Commit 13a1f735 authored by Yuxin Wu's avatar Yuxin Wu

auto set logging directory

parent a8dfad63
......@@ -87,8 +87,6 @@ class Model(DQNModel):
def get_config():
logger.auto_set_dir()
M = Model()
expreplay = ExpReplay(
predictor_io_names=(['state'], ['Qvalue']),
......@@ -164,6 +162,9 @@ if __name__ == '__main__':
elif args.task == 'eval':
eval_model_multithread(cfg, EVAL_EPISODE, get_player)
else:
logger.set_logger_dir(
'train_log/DQN-{}'.format(
os.path.basename(ROM_FILE).split('.')[0]))
config = get_config()
if args.load:
config.session_init = SaverRestore(args.load)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment