在训练的时候glops计算量不显示是怎么回事

[zhangshoulong]

在训练的时候glops计算量不显示是怎么回事

[icey-zhang]

用test.py文件中的thop库可以测试参数量和GFLOPs。解开注释就可以。
# import thop
# input_image_size = opt.img_size
# input_image = torch.randn(1, 3, input_image_size, input_image_size).to(device)
# flops, params = thop.profile(model, inputs=(input_image,input_image,input_mode), verbose=False)
# print('Params: %.4fM'%(params/1e6))
# print('FLOPs: %.2fG'%(flops/1e9))

[zhangshoulong]

感谢

[leileilei2000]

用 test.py 文件中的thop库可以测试参数量和GFLOPs。解开注释就可以。 # import thop # input_image_size = opt.img_size # input_image = torch.randn（1， 3， input_image_size， input_image_size）.to（device） # flops， params = thop.profile（model， inputs=（input_image，input_image，input_mode）， verbose=False） # print（'Params： %.4fM'%（params/1e6）） # print（'FLOPs： %.2fG'%（flops/1e9））

为什么前后输出的参数不一样呢？