DeepSpeed
1eb9b218 - [configure_distributed_model] improve assert (#1053)

Comment changes are shownComment changes are hidden
Commit
4 years ago
[configure_distributed_model] improve assert (#1053) * [configure_distributed_model] improve assert This PR changes the 2 asserts to actually print the names of the params that are wrong. e.g.: ``` fp16 is enabled but the following parameters have dtype that is not fp16: wav2vec2.masked_spec_embed ``` * style Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Author
Parents
  • deepspeed/runtime
    • File
      engine.py