okankop/Efficient-3DCNNs

can’t calculate_FLOP

Closed this issue · 1 comments

i installed pytorch 1.4.
when i instead c3d to resnet.resnet50,it say:"profile() got an unexpected keyword argument 'input_size'" .
i wanna calculate 3D model‘s flops.May i make a mistake at somewhere?

Hi @shxijack, could you please try re-cloning the repo. Because in the thop library profile function expects to take 'input_size' as input. You can check it out at line25.