1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

Enable creating custom tensorflow heads, embedders, and middleware. (#135)

Allowing components to have a path property.
This commit is contained in:
Ryan Peach
2018-12-05 04:40:06 -05:00
committed by Gal Leibovich
parent 3c58ed740b
commit 9e66bb653e
4 changed files with 16 additions and 5 deletions

View File

@@ -31,6 +31,11 @@ class HeadParameters(NetworkComponentParameters):
self.loss_weight = loss_weight
self.parameterized_class_name = parameterized_class_name
@property
def path(self):
return 'rl_coach.architectures.tensorflow_components.heads:' + self.parameterized_class_name
class PPOHeadParameters(HeadParameters):
def __init__(self, activation_function: str ='tanh', name: str='ppo_head_params',