DeepSpeed
DeepSpeed copied to clipboard
some fix in flops_profiler
-
Tensor.bmmmissed in_patch_tensor_methodsfunction - missed funtions in
_reload_functionalsand_reload_tensor_methodsfunctions -
torch.mmandtorch.Tensor.mmwill have same__name__inwrapFunc, my suggustion is use__str__instead. - add
torch.baddbmm
@cli99
@lucasleesw, thanks for the PR. Can you please use this to resolve the formatting issues
Hi @lucasleesw , thanks for the PR. Is there a reason you use str instead of str()?
Hi @lucasleesw , thanks for the PR. Is there a reason you use str instead of str()?
Nothing specific, just choose another attribute instead of __name__.