gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/architectures/embedder_parameters.py
View File

@@ -18,6 +18,7 @@ from typing import List, Union
from rl_coach.base_parameters import EmbedderScheme, NetworkComponentParameters
MOD_NAMES = {'image': 'ImageEmbedder', 'vector': 'VectorEmbedder', 'tensor': 'TensorEmbedder'}
class InputEmbedderParameters(NetworkComponentParameters):
def __init__(self, activation_function: str='relu', scheme: Union[List, EmbedderScheme]=EmbedderScheme.Medium,
@@ -39,3 +40,6 @@ class InputEmbedderParameters(NetworkComponentParameters):
self.input_clipping = input_clipping
self.name = name
self.is_training = is_training
def path(self, emb_type):
return 'rl_coach.architectures.tensorflow_components.embedders:' + MOD_NAMES[emb_type]