Skip to content

Commit 1b3183c

Browse files
authored
871 Add AMP tutorial and compare performance (#879)
* [DLMED] add AMP notebook * [DLMED] train the model with/without AMP * [DLMED] add memory usage screenshots
1 parent 461255a commit 1b3183c

File tree

3 files changed

+424
-0
lines changed

3 files changed

+424
-0
lines changed

automatic_mixed_precision.ipynb

Lines changed: 424 additions & 0 deletions
Large diffs are not rendered by default.

images/mem_amp_off.png

59.8 KB
Loading

images/mem_amp_on.png

59.8 KB
Loading

0 commit comments

Comments
 (0)