search

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework
MIT License
2.83k stars 306 forks source link

Is it able to count also for layers from torch_geometric? #136

Open harborsarah opened 6 months ago

harborsarah commented 6 months ago

Dear authors,

Since my model contains also layers from torch_geometric, and my model has multiple inputs. Some of them have a predefined size but one input is the point cloud, which does not have a predefined shape from frame to frame. Instead, it is generated via torch_geometric.data.Batch.from_data_list() etc.

I was woundering, if this package can also handel this?

Thanks a lot for your reply.