Pytorch-StNet
Pytorch-StNet copied to clipboard
Questions about get the FLOPs of the model
I wrote my own code to get the FLOPs of StNet and MFNet, but the results I got is different from the results shown in the paper, whether you can show me the code of computing the FLOPs, especially for the nn.Conv3d layer.
Thank you
Hi. Unfortunately I'm unfamiliar with any FLOPS estimation tool. But as I am not aware of all the hyperparameters used by the team that wrote the paper, it's probable that our FLOPS diverge.