![]() ![]() predict - Returns predictions (with metrics if labels are available) on a test set.evaluate - Runs an evaluation loop and returns metrics.prediction_step - Performs an evaluation/test step.training_step - Performs a training step.compute_loss - Computes the loss on a batch of training inputs.create_scheduler - Sets up the learning rate scheduler if it wasn’t passed at init.create_optimizer - Sets up the optimizer if it wasn’t passed at init.Note, that you can also subclass or override the create_optimizer and create_scheduler methods create_optimizer_and_scheduler - Sets up the optimizer and learning rate scheduler if they were not passed at. ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |