2025年12月5日
计算 PyTorch 模型的参数量和 FLOPs
摘要： from mmcv.cnn import get_model_complexity_info # from ptflops import get_model_complexity_info flops, params = get_model_complexity_info( model, input 阅读全文
posted @ 2025-12-05 11:20 zhuangzhongxu 阅读(9) 评论(0) 推荐(0)