AttributeError: ‘SRmodel‘ object has no attribute ‘apply‘

最新推荐文章于 2024-07-24 17:27:32 发布

狗庄欺人太甚

最新推荐文章于 2024-07-24 17:27:32 发布

阅读量464

点赞数 7

文章标签： java 前端 linux

本文链接：https://blog.csdn.net/weixin_48262500/article/details/135369298

版权

文章概述了对两个模型（可能是一个生成器模型，model_G）的性能评估，包括计算激活的数量、卷积层的数量、FLOPs（浮点运算次数）以及参数总数，以衡量其在IT技术中的资源消耗和效率。

摘要由CSDN通过智能技术生成



from utils.utils_modelsummary import get_model_activation, get_model_flops
input_dim = (3, 256, 256)  # set the input dimension
activations, num_conv2d = get_model_activation(model, input_dim)
logger.info('{:>16s} : {:<.4f} [M]'.format('#Activations', activations/10**6))
logger.info('{:>16s} : {:<d}'.format('#Conv2d', num_conv2d))
flops = get_model_flops(model, input_dim, False)
logger.info('{:>16s} : {:<.4f} [G]'.format('FLOPs', flops/10**9))
num_parameters = sum(map(lambda x: x.numel(), model.parameters()))
logger.info('{:>16s} : {:<.4f} [M]'.format('#Params', num_parameters/10**6))

改进后



from utils.utils_modelsummary import get_model_activation, get_model_flops
model_G = model.netG.module
input_dim = (3, 256, 256)  # set the input dimension
activations, num_conv2d = get_model_activation(model, input_dim)
logger.info('{:>16s} : {:<.4f} [M]'.format('#Activations', activations/10**6))
logger.info('{:>16s} : {:<d}'.format('#Conv2d', num_conv2d))
flops = get_model_flops(model, input_dim, False)
logger.info('{:>16s} : {:<.4f} [G]'.format('FLOPs', flops/10**9))
num_parameters = sum(map(lambda x: x.numel(), model.parameters()))
logger.info('{:>16s} : {:<.4f} [M]'.format('#Params', num_parameters/10**6))

狗庄欺人太甚

关注

7
点赞
踩
6

收藏

觉得还不错? 一键收藏
0
评论
AttributeError: ‘SRmodel‘ object has no attribute ‘apply‘

【代码】AttributeError: ‘SRmodel‘ object has no attribute ‘apply‘
复制链接

扫一扫