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

Comment changes are shownComment changes are hidden
Commit
1 year 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
  • deepspeed/profiling/flops_profiler
    • File
      profiler.py