Searched refs:eval_dataloader (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/benchmarks/dynamo/ |
| D | training_loss.py | 41 eval_dataloader = DataLoader(small_eval_dataset, batch_size=batch_size)
43 return train_dataloader, eval_dataloader
56 backend, train_dataloader, eval_dataloader, model, optimizer, num_epochs, evaluation argument
84 for batch in eval_dataloader:
149 train_dataloader, eval_dataloader = data_processing(
164 eval_dataloader,
174 eval_dataloader,
|