2020年2月26日
使用torchsummary打印torch模型结构，包括每层名字以及形状
摘要： import models from torchsummary import summary model = models.resnet50(pretrained=False, num_classes=7, scale=1) print(model) summary(model, (3, 640, 阅读全文
posted @ 2020-02-26 19:46 you-wh 阅读(3818) 评论(0) 推荐(0) 编辑
Fork me on GitHub