Skip to content
This repository has been archived by the owner on Dec 16, 2022. It is now read-only.

Enable AMP with Apex #3866

Merged
12 commits merged into from
Mar 2, 2020
Merged

Enable AMP with Apex #3866

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
f49ec6b
Enable AMP with Apex
JohnGiorgi Feb 27, 2020
164a6e7
Add opt_level to docstring
JohnGiorgi Feb 27, 2020
3d3abaf
Move apex import
JohnGiorgi Feb 27, 2020
eb69fde
Refactors rescale_gradients
JohnGiorgi Feb 28, 2020
f765699
Add apex to dev requirements
JohnGiorgi Feb 28, 2020
e085648
Add test to check amp
JohnGiorgi Feb 28, 2020
609352a
Add missing import
JohnGiorgi Feb 28, 2020
e3a29c6
Error if apex not installed but opt_level not None
JohnGiorgi Feb 28, 2020
7f2fce0
formatted with black
JohnGiorgi Feb 28, 2020
5941b95
linted with flake8
JohnGiorgi Feb 28, 2020
5fc9654
removes importlib from trainer_test
JohnGiorgi Mar 2, 2020
3b95273
Merge branch 'master' into master
DeNeutoy Mar 2, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions dev-requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,9 @@ matplotlib>=2.2.3
# Required to run sanic tests
aiohttp

# Required for automatic mixed precision (AMP) training
git+https://github.com/NVIDIA/apex.git@master

Comment on lines +28 to +30
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Okay, as discussed, I have added apex as a dev requirement

#### DOC-RELATED PACKAGES ####

# YAML manipulation
Expand Down