Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
conda install dnachun::ultralytics-thop