Skip to content
Snippets Groups Projects
Unverified Commit a4331cec authored by le-horizon's avatar le-horizon Committed by GitHub
Browse files

fix bug where epsilon_greedy is not passed to ppg_network_forward (#1447)


(cherry picked from commit 11318a98b839b37575052b2db233615b87cc91db)

Co-authored-by: default avatarLe Horizon <le.zhao+git@horizon.ai>
parent f22c7a61
No related branches found
No related tags found
No related merge requests found
......@@ -199,8 +199,11 @@ class PPGAlgorithm(OffPolicyAlgorithm):
return self._loss(info)
def predict_step(self, inputs: TimeStep, state):
return ppg_network_forward(self._network, inputs, state,
self._predict_step_epsilon_greedy)
return ppg_network_forward(
self._network,
inputs,
state,
epsilon_greedy=self._predict_step_epsilon_greedy)
def after_train_iter(self, experience, info: PPGTrainInfo):
"""Run auxiliary update if conditions are met
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment