model=torch.nn.DataParallel(model_shuffle)# 調(diào)用shufflenet2 模型,該模型為自己定義的 flop, para=profile(model, input_size=(1,3,224,224),) print("%.2fM"%(flop/1e6),"%.2fM"%(para/1e6)) 更多細(xì)節(jié),可參考thop GitHub鏈接:https://g
www.dbjr.com.cn/article/2068...htm 2025-6-3