Skip to content

Full PyTorch experiment tracking with support for DP/DDP & APEX AMP
Compare
Choose a tag to compare
@mv1388 mv1388 released this 30 Mar 20:50
· 1363 commits to master since this release
1.0
355e4ac
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

AIToolbox training setup with automatic experiment tracking.

With this release, TrainLoop supports in addition to the single GPU and DataParallel training also the DistributedDataParallel multi-GPU setting.
Furthermore this release also fully enables mixed-precision training via Nvidia APEX for both single GPU as well as multi-GPU training settings.

Assets 3