corey-nm
corey-nm
Currently there is no way to prevent AutoBackend from fusing a model together. If the model has been already fused or quantized elsewhere, then AutoBackend doesn't need to do this....
This enables easily customizing the data that is saved in the checkpoints by subclassing from `network.Trainer` and overwriting build_checkpoint.
- Renames `BaseManager.phase` -> `BaseManager.phase_at_end_of` - Clarifies behavior - Integrates saving checkpoints based on phases into torchvision # Test Plan Ran the following recipe: ```yaml version: 1.1.0 training_modifiers: - !EpochRangeModifier...
**NOTE: this PR is targeting `torchvision-phases` branch, not main!** Similar to #1499, this updates checkpoint saving logic to save checkpoints with the phase the model is in appended to the...