log - Logs information on the various objects watching training.get_test_dataloader - Creates the test DataLoader.get_eval_dataloader - Creates the evaluation DataLoader.get_train_dataloader - Creates the training DataLoader.To inject custom behavior you can subclass them and override the following methods: The Trainer contains the basic training loop which supports the above features. The API supports distributed training on multiple GPUs/TPUs, mixed precision through NVIDIA Apex and Native AMP for PyTorch. It’s used in most of the example scripts.īefore instantiating your Trainer, create a TrainingArguments to access all the points of customization during training. The Trainer class provides an API for feature-complete training in PyTorch for most standard use cases.
0 kommentar(er)
