Update ACVP

2026-09-20 12:20:38 +08:00 · 2018-04-10 11:32:59 -06:00
parent af1cc11669
commit 669c366bc0
3 changed files with 6 additions and 3 deletions
@@ -50,6 +50,7 @@ def generate_dataset(game, a2c_model, prefix):
    config.gradient_clip = 0.5
    config.logger = Logger('./log', logger, skip=True)
    agent = A2CAgent(config)
+    agent.close()

    agent.load(a2c_model)
    task = PixelAtari(game, frame_skip=4, history_length=4, log_dir=None, dataset=True)