falta 1 argumento posicional requerido, en el resumen del modelo
Intenté imprimir el resumen del modelo SinGAN, pero obtuve un error que es:
Este es el código:
def init_models(opt):
# Inicialización del generador:
netG = models.GeneratorConcatSkip2CleanAdd(opt).to(opt.device)
netG.apply(models.weights_init)
if opt.netG != '':
netG.load_state_dict(torch.load(opt.netG))
summary(netG, input_size=(3, 201, 256))
print(netG)
# Inicialización del discriminador:
netD = models.WDiscriminator(opt).to(opt.device)
netD.apply(models.weights_init)
if opt.netD != '':
netD.load_state_dict(torch.load(opt.netD))
print(netD)
return netD, netG
El problema ocurre cuando añado esta línea:
summary(netG, input_size=(3, 201, 256))
Y obtengo el código completo de aquí.
Entonces, ¿está mal en mi forma de hacerlo? ¿Debería usar una variable diferente como modelo?
davy.ai
El mensaje de error sugiere que no existe un módulo llamado ‘thop’. THOP es un paquete de PyTorch para estimar los costos computacionales de los modelos de PyTorch. Parece que no está instalado en tu entorno.
Debes instalar thop ejecutando el siguiente comando en tu terminal:
Una vez instalado, la función
summarydebería funcionar correctamente.Tu enfoque parece correcto y estás utilizando la variable correcta
netGcomo el modelo. El problema está en la dependencia faltante necesaria para calcular el resumen.Nota: Si estás utilizando un cuaderno de Google Colab, es posible que debas reiniciar el kernel después de instalar thop para que los cambios surtan efecto.