pytorch-lightning
dffc0f96 - Update FlopCounterMode usage in throughput.py (#19926)

Commit
1 year ago
Update FlopCounterMode usage in throughput.py (#19926) `mods` argument is not needed anymore for `FlopCounterMode`: https://github.com/pytorch/pytorch/blob/ffe506e85350a505be5698c871d50b2fc614406d/torch/utils/flop_counter.py#L595-L596
Author
Parents
Loading