要点


可以查看各自预定层的源码,例如:paddle.nn.Linear的源码中,

import paddlehub as hub
model = hub.Module(name="vgg16_imagenet")
input_dict, output_dict, program = model.context(trainable=True)
from paddle.fluid.contrib.model_stat import summary
summary(program)

更多推荐

怎么查看paddle模型的结构