2024年10月30日
展示模型概要summary，查看模型结构工具Netron
摘要： 推荐torchinfo，官网地址 TylerYep/torchinfo: View model summaries in PyTorch! torchsummary、torch-summary都用过，还是torchinfo更准，尤其是显示yolo模型的时候。 展示形式如下： 对于一般模型，如resn 阅读全文
posted @ 2024-10-30 17:02 夕西行 阅读(17) 评论(0) 推荐(0) 编辑