model=resnet50() flops, params=profile(model, input_size=(1,3,224,224))# profile(模型,輸入數(shù)據(jù)) 對于自己構(gòu)建的函數(shù)也一樣,例如shuffleNetV2 1 2 3 4 5 6 7 8 fromthopimportprofile fromutils.ShuffleNetV2importshufflenetv2# 導(dǎo)入shufflenet2 模塊 ...
www.dbjr.com.cn/article/2068...htm 2025-6-12