你好,请问params为什么我用 summary测试出来small的6.38m,large为19.76m,请问是什么原因呢?
zclyaya opened this issue · 1 comments
zclyaya commented
chaochen1998 commented
你看前面的名称“Params size (MB) : 19.76”,意思是你输入的图片(3 x width x height
)在某个精度下的内存大小为19.76MB
,前面的Trainable params: 5,178,732
意思是可训练的参数数量为5178732
。