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:
committed by
Gal Leibovich
parent
3c58ed740b
commit
9e66bb653e
@@ -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',
|
||||
|
||||
Reference in New Issue
Block a user