Searched defs:train_dataloader (Results 1 – 2 of 2) sorted by relevance
56 backend, train_dataloader, eval_dataloader, model, optimizer, num_epochs, evaluation argument
91 train_dataloader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffl… variable