pytorch imprime el tamaño total de los parámetros

#model 为你要打印的参数模型
print("Total number of paramerters in networks is {}  ".format(sum(x.numel() for x in model.parameters())))

Supongo que te gusta

Origin blog.csdn.net/qq_43360777/article/details/106382789
Recomendado
Clasificación