tests/distributed/amp_master_params/amp_master_params.py · 6f7a8b3958b1e029e81e48436f962715ea0b6e1c · OpenDAS / apex

"vscode:/vscode.git/clone" did not exist on "6845988807e458e8e0eb406a2739080f52c267bd"

rohithkrn authored Jun 03, 2020

* bfloat16 support for apex DDP

* enable mgpu tests for fp16 and bf16

* update Dockerfile

b0c7d09f

amp_master_params.py 2.8 KB