some fix in flops_profiler (#2068)
* bugs in profiler: 1. Tensor.bmm missed in _patch_tensor_methods function 2. missed funtions in _reload_functionals and _reload_tensor_methods functions 3. torch.mm and torch.Tensor.mm will have same __name__ in wrapFunc, my suggustion is use __str__ instead. * formatting --------- Co-authored-by: NOlatunji Ruwase <olruwase@microsoft.com> Co-authored-by: NCheng Li <pistasable@gmail.com>
Showing
想要评论请 注册 或 登录