|
def | model_build_fun (self, model, forward_only=False, loss_scale=None) |
|
def | forward_model_build_fun (self, model, loss_scale=None) |
|
def | norm_clipped_grad_update (self, model, scope) |
|
def | norm_clipped_sparse_grad_update (self, model, scope) |
|
def | total_loss_scalar (self) |
|
def | __init__ (self, model_params, source_vocab_size, target_vocab_size, num_gpus=1, num_cpus=1) |
|
def | __enter__ (self) |
|
def | __exit__ (self, exc_type, exc_value, traceback) |
|
def | initialize_from_scratch (self) |
|
def | get_current_step (self) |
|
def | inc_current_step (self) |
|
def | step (self, batch, forward_only) |
|
def | save (self, checkpoint_path_prefix, current_step) |
|
|
| num_gpus |
|
| model |
|
| forward_net |
|
| learning_rate |
|
| global_step |
|
| start_time |
|
| encoder_embeddings |
|
| decoder_embeddings |
|
| model_params |
|
| encoder_type |
|
| encoder_params |
|
| source_vocab_size |
|
| target_vocab_size |
|
| num_cpus |
|
| batch_size |
|
Definition at line 99 of file train.py.
The documentation for this class was generated from the following file: