分享至
大家好,最近在自學機器學習,真的很菜但又沒有老師只好上來問Q_Q想請問如果我有65張圖,batch是64的話,跑一個epoch,iteration是1次還是2次?iteration可以為浮點數嗎?另外,YOLOv4建議的max_batch(我理解為iteration次數)為class乘2或6000,但v7好像把這個超參數拿掉、剩epoch,那還需要用epoch回推iteration,讓iteration落在class乘2或6000嗎?還是從訓練跟驗證的loss決定何時停止呢?
已邀請的邦友 {{ invite_list.length }}/5
iteration 是整數, 你的設定應該算為 2
參考這個https://medium.com/%E4%BA%BA%E5%B7%A5%E6%99%BA%E6%85%A7-%E5%80%92%E5%BA%95%E6%9C%89%E5%A4%9A%E6%99%BA%E6%85%A7/epoch-batch-size-iteration-learning-rate-b62bf6334c49
YOLOv7 仍有 batch-size
好的,了解了。感謝您的回答~(有看過那篇文章也爬過很多跟iteration有關的文,但好像大家的舉例都是整數,所以之前不太了解~)
IT邦幫忙