From 943e41ba587015ecc8b062a40b08eb87f7b05dd5 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Fri, 16 Feb 2018 13:36:14 -0500 Subject: [PATCH] fix nec_agent --- agents/nec_agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/agents/nec_agent.py b/agents/nec_agent.py index 2e06338..c123458 100644 --- a/agents/nec_agent.py +++ b/agents/nec_agent.py @@ -18,6 +18,7 @@ import numpy as np from agents.value_optimization_agent import ValueOptimizationAgent from logger import screen +from utils import RunPhase # Neural Episodic Control - https://arxiv.org/pdf/1703.01988.pdf @@ -40,7 +41,7 @@ class NECAgent(ValueOptimizationAgent): screen.log_title("Finished collecting initial entries in DND. Starting to train network...") current_states, next_states, actions, rewards, game_overs, total_return = self.extract_batch(batch) - result = self.main_network.train_and_sync_networks([current_states, actions], total_return) + result = self.main_network.train_and_sync_networks(current_states, total_return) total_loss = result[0] return total_loss