原来用hapi定义模型时,在动态图模式下直接这样写就ok的
dense_model = DenseNet(num_classes=CLASS_DIM) model = paddle.Model(dense_model) print("使用稠密块的模型的结构:") model.summary((-1, 3, 224, 224))
我刚反应过来 哈哈
请问是用哪个api呀。。。我是菜鸟,用summary打印不出来呜呜呜
paddle.Model.summary() 或 paddle.summary()
文档:
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/hapi/model/Model_cn.html#summary-input-size-none-batch-size-none-dtype-none
我刚反应过来 哈哈
请问是用哪个api呀。。。我是菜鸟,用summary打印不出来呜呜呜
paddle.Model.summary() 或 paddle.summary()
文档:
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/hapi/model/Model_cn.html#summary-input-size-none-batch-size-none-dtype-none