DeepSpeed
b5750b64 - fix upsample flops compute by skipping unused kargs (#2773)

Comment changes are shownComment changes are hidden
Commit
2 years ago
fix upsample flops compute by skipping unused kargs (#2773) * fix upsample flops compute by skipping unused kargs * fix format
Author
Parents
  • deepspeed/profiling/flops_profiler
    • File
      profiler.py