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

@@ -32,6 +32,10 @@ class MiddlewareParameters(NetworkComponentParameters):
self.is_training = is_training
self.parameterized_class_name = parameterized_class_name
@property
def path(self):
return 'rl_coach.architectures.tensorflow_components.middlewares:' + self.parameterized_class_name
class FCMiddlewareParameters(MiddlewareParameters):
def __init__(self, activation_function='relu',