調MURA數據集keras CNN模型筆記

永遠不要再把sigmoid寫錯成softmax cnn層最後加三個fc層比加1個效果要好得多 若是參數太多,能夠在最後一個cnn層後面加一個maxpooling整個圖 要記得指定data_format=「channel_first",此時batch_norm層要指定axis=1 20w參數的模型在GTX1050上用 tf2 勉強能跑的 tf2 性能比 pytorch好太多了,首先在顯存和內存的管理
相關文章
相關標籤/搜索