EADST

Check Torch Model Structure, Flops, and Params

from torchinfo import summary
images.tensors.size() # (3, 640, 640)
summary(torch_model, images.tensors.size())

from thop import profile
flops, params = profile(torch_model, inputs=[images], verbose=False)

from ptflops import get_model_complexity_info
flops, params = get_model_complexity_info(torch_model, (3, 640, 640), as_strings=True, print_per_layer_stat=True)
flops, params = get_model_complexity_info(torch_model, [images.tensors.size()], as_strings=True, print_per_layer_stat=True)