diff --git a/torchtnt/utils/flops.py b/torchtnt/utils/flops.py index a7a404040e..aec8eb5737 100644 --- a/torchtnt/utils/flops.py +++ b/torchtnt/utils/flops.py @@ -179,6 +179,10 @@ class FlopTensorDispatchMode(TorchDispatchMode): Flop count implementation based on https://dev-discuss.pytorch.org/t/the-ideal-pytorch-flop-counter-with-torch-dispatch/505 + Note that not all operators are supported. + Unsupported operations will be logged as debug messages but will not contribute + to the FLOP count. + Examples:: >>> import copy