You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Pytorch 1.7.0: Currently Pytorch 1.7.0 or higher is supported.
Metrics visualization: Support Visdom for real-time loss visualization during training.
Automatic mixed precision training: Train faster
with AMP and less GPU memory
on NVIDIA tensor cores
Pretrained models provided: Load the pretrained weights.
GPU/CPU support for inference: Runs on GPU or CPU in inference time.
Intelligent training procedure: The state of the model, optimizer, learning rate scheduler and so on... You can
stop your training and resume training exactly from the checkpoint.