徹底解決keras model.summary()或者layer.count_params()權重參數個數爲負數問題

使用keras model.summary()或者layer.count_params()權重參數個數會出現爲負數的情況,出現情況一般在全連接中,比如使用不經過圖像壓縮的(1024,1024,3)大小的圖片經過Densenet,在全連接層中共有權重參數2147487744 = 524288*4096 + 4096,但因爲默認使用的是np.int32,所以在返回值時表示爲: 所以我們需要修改kera
相關文章
相關標籤/搜索