2021年12月18日
TorchSummary无法载入Dict数据类型解决方法
摘要： 前言 torchsummary是一个比较不错的评价网络数据结构的包，但是目前torchsummary只支持网络输入和输出为torch.Tensor类型的网络，在对一些较为复杂的网络模型中，载入的可能并不一定为tensor类型，也有可能是list或者dict类型的数据。 为了可以支持自己的网络结构，为 阅读全文
posted @ 2021-12-18 14:28 kenvision 阅读(873) 评论(0) 推荐(0) 编辑