摘要: 最近在研究模型的计算量,发现Pytorch有库可以直接计算模型的计算量,所以需要一个一个Keras和Tensorflow可以用的,直接把Model接入到函数中,print一下就可以计算出FLOPs FLOPS:注意全大写,是floating point operations per second的缩 阅读全文
posted @ 2022-04-12 21:43 咖啡陪你 阅读(1952) 评论(0) 推荐(0)