num_workers:表示開啟多少個執行緒數去載入你的資料,預設為0,代表只使用主程序。 collate_fn:表示合併樣本列表以形成小批量的Tensor物件。 pin_memory:表示要將load ...
確定! 回上一頁