Searched defs:eval_dataset (Results 1 – 3 of 3) sorted by relevance
61 def train_and_eval(device_id, epoch_size, model, dataset, loss_cb, eval_dataset): argument
145 eval_dataset = create_dataset(1, training=False) variable