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

Add docstring for architecture (#47)

- Removed get_model() from architecture because it is only implementation detail of architecture.
This commit is contained in:
Sina Afrooze
2018-10-30 02:02:37 -07:00
committed by Gal Leibovich
parent 324c67d614
commit 2046358ab0
2 changed files with 146 additions and 15 deletions

View File

@@ -146,6 +146,14 @@ class TensorFlowArchitecture(Architecture):
# set the fetches for training
self._set_initial_fetch_list()
def get_model(self) -> None:
"""
Constructs the model using `network_parameters` and sets `input_embedders`, `middleware`,
`output_heads`, `outputs`, `losses`, `total_loss`, `adaptive_learning_rate_scheme`,
`current_learning_rate`, and `optimizer`
"""
raise NotImplementedError
def _set_initial_fetch_list(self):
"""
Create an initial list of tensors to fetch in each training iteration