2024年8月28日
pytorch统计模型计算强度
摘要： 计算强度 = 运算数 / 访存数 运算数有很多库可以算，例如thop的profile from thop import profile model = torchvision.models.resnet18(weights=None) flops, params = profile(model, i 阅读全文
posted @ 2024-08-28 20:40 心比天高xzh 阅读(106) 评论(0) 推荐(0)