mirror of
https://github.com/gryf/coach.git
synced 2025-12-17 11:10:20 +01:00
bug fix for clipped ppo for discrete controls
This commit is contained in:
@@ -194,7 +194,9 @@ class ClippedPPOAgent(ActorCriticAgent):
|
||||
for input_index, input in enumerate(old_policy_distribution):
|
||||
inputs['output_1_{}'.format(input_index + 1)] = input
|
||||
|
||||
inputs['output_1_3'] = self.ap.algorithm.clipping_decay_schedule.current_value
|
||||
# update the clipping decay schedule value
|
||||
inputs['output_1_{}'.format(len(old_policy_distribution)+1)] = \
|
||||
self.ap.algorithm.clipping_decay_schedule.current_value
|
||||
|
||||
total_loss, losses, unclipped_grads, fetch_result = \
|
||||
self.networks['main'].train_and_sync_networks(
|
||||
|
||||
Reference in New Issue
Block a user