search

hmorimitsu / ptlflow

PyTorch Lightning Optical Flow models, scripts, and pretrained weights.
Apache License 2.0
250 stars 33 forks source link

FLOPS Measurement #25

Closed GotG closed 2 years ago

GotG commented 2 years ago

Hello, quick question please:

Do you have any suggestion as to how to measure the FLOPS of the models?

Thank you!

hmorimitsu commented 2 years ago

Hi, I have seen some libraries that do this in pytorch, but I have never used any of them myself. So I don't have any recommendations, sorry.