要点
可以查看各自预定层的源码,例如: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模型的结构
发布评论