Skip to content

Commit

Permalink
Merge pull request #635 from mv1388/readme-fix
Browse files Browse the repository at this point in the history
Readme Fix
  • Loading branch information
mv1388 committed Oct 31, 2020
2 parents 10ee796 + c8eceeb commit 5d94645
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -180,7 +180,7 @@ TrainLoop(
).fit(num_epochs=10)
```

Check out a full [Apex AMP training example](https://github.com/mv1388/aitoolbox/blob/master/examples/amp_training/single_GPU_training.py).
Check out a full [AMP single-GPU training example](https://github.com/mv1388/aitoolbox/blob/master/examples/amp_training/single_GPU_training.py).

### Multi-GPU DDP mixed precision training
When training in the multi-GPU setting, the setup is mostly the same as in the single-GPU.
Expand All @@ -201,7 +201,7 @@ TrainLoop(
).fit(num_epochs=10)
```

Check out a full [Apex AMP DistributedDataParallel training example](https://github.com/mv1388/aitoolbox/blob/master/examples/amp_training/mutli_GPU_training.py).
Check out a full [AMP multi-GPU DistributedDataParallel training example](https://github.com/mv1388/aitoolbox/blob/master/examples/amp_training/mutli_GPU_training.py).


## Model
Expand Down

0 comments on commit 5d94645

Please sign in to comment.