Question: How can I calculate the flops and params of each model?

unit8co / darts

A python library for user-friendly forecasting and anomaly detection on time series.

https://unit8co.github.io/darts/

Apache License 2.0

7.92k stars 859 forks source link

Closed DomineeringDragon closed 2 years ago

DomineeringDragon commented 2 years ago

I want to calculate flops and params of each model, I tried thop and ptflops in the NBEATs Model but nothing can be calculate....

dennisbader commented 2 years ago

We currently don't have a way to calculate flops for our TorchForecastingModels. Maybe PyTorch Lightning will add this in the future, see here.

The number of model paramaters is printed in the model summary when calling model.fit().