需要注意的是,如果dataset的大小並不能被batch_size整除,則dataloader中最後一個batch可能比實際的batch_size要小。例如,對於1001個樣本,batch_size的 ...
確定! 回上一頁