pytorch_pfn_extras.training.extensions#
Functions
Returns an extension to record the learning rate. |
|
Returns an extension to continuously record a value. |
|
Returns a trainer extension to take snapshots of the trainer. |
|
Returns an extension to take snapshots of a given object. |
Classes
|
|
Extension traces the best value of a specific key in the observation. |
|
|
An extension to evaluate models on a validation set in a distributed training setup. |
An extension to evaluate models on a validation set. |
|
An extension to raise RuntimeError if parameters and its gradients contain NaN or Inf. |
|
Trainer extension to adjust the learning rate using PyTorch's learning rate scheduler. |
|
An extension to output the accumulated results to a log file. |
|
Extension traces the maximum value of a specific key in the observation. |
|
Calculates micro-average ratio. |
|
Extension traces the maximum value of a specific key in the observation. |
|
|
An extension to report parameter statistics. |
An extension to output plots. |
|
An extension to print the accumulated results. |
|
alias of |
|
Writes the profile results to a file. |
|
An extension to print a progress bar and recent training status. |
|
alias of |
|
An extension to communicate with Slack. |
|
An extension to communicate with Slack using Incoming Webhook. |
|
An enumeration. |
|
|
|
Writes the profile timeline to a file. |
|
|
|
|
An extension to plot statistics for |
Modules
|