batch_size=BATCH_SIZE, # mini batch size. 17. shuffle=True, # 要不要打乱数据(打乱比较好). 18. num_workers=2, # 多线程来读数据.
確定! 回上一頁