DeepSpeed
c1c1d249 - fix retrieval of out_channels in _conv_trans_flops_compute (#3834)

Commit
2 years ago
fix retrieval of out_channels in _conv_trans_flops_compute (#3834) Co-authored-by: Pinstripe Potoroo <pinstripe-potoroo@users.github.com> Co-authored-by: Cheng Li <pistasable@gmail.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Parents
Loading