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
@@ -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',
|
||||
|
||||
Reference in New Issue
Block a user