a trainer with no added functionality More...
Inheritance diagram for nabu.neuralnetworks.trainers.standard_trainer.StandardTrainer:
Public Member Functions | |
| def | aditional_loss (self) |
| add an aditional loss More... | |
| def | chief_only_hooks (self, outputs) |
| add hooks only for the chief worker More... | |
| def | hooks (self, outputs) |
| add hooks for the session More... | |
 Public Member Functions inherited from nabu.neuralnetworks.trainers.trainer.Trainer | |
| def | __init__ (self, conf, dataconf, modelconf, evaluatorconf, expdir, server, task_index) |
| NnetTrainer constructor, creates the training graph. More... | |
| def | train (self, testing=False) |
| train the model More... | |
| def | chief_only_hooks (self, outputs) |
| add hooks only for the chief worker More... | |
| def | hooks (self, outputs) |
| add hooks for the session More... | |
| def | aditional_loss (self) |
| add an aditional loss More... | |
Additional Inherited Members | |
| Public Attributes inherited from nabu.neuralnetworks.trainers.trainer.Trainer | |
| conf | |
| dataconf | |
| evaluatorconf | |
| expdir | |
| server | |
| task_index | |
| model | |
Detailed Description
a trainer with no added functionality
Member Function Documentation
§ aditional_loss()
| def nabu.neuralnetworks.trainers.standard_trainer.StandardTrainer.aditional_loss | ( | self | ) |
add an aditional loss
- Returns
- the aditional loss or None
§ chief_only_hooks()
| def nabu.neuralnetworks.trainers.standard_trainer.StandardTrainer.chief_only_hooks | ( | self, | |
| outputs | |||
| ) |
add hooks only for the chief worker
- Parameters
-
outputs the outputs generated by the create graph method
- Returns
- a list of hooks
§ hooks()
| def nabu.neuralnetworks.trainers.standard_trainer.StandardTrainer.hooks | ( | self, | |
| outputs | |||
| ) |
add hooks for the session
- Parameters
-
outputs the outputs generated by the create graph method
- Returns
- a list of hooks
The documentation for this class was generated from the following file:
