Searched defs:train_data_loader (Results 1 – 2 of 2) sorted by relevance
/dports/misc/py-gluonnlp/gluon-nlp-0.10.0/scripts/natural_language_inference/ |
H A D | main.py | 105 def train_model(model, train_data_loader, val_data_loader, embedding, ctx, args): argument
|
/dports/misc/py-gluonnlp/gluon-nlp-0.10.0/docs/examples/machine_translation/ |
H A D | utils.py | 97 def train_one_epoch(epoch_id, model, train_data_loader, trainer, label_smoothing, loss_function, gr… argument
|