DataLoader 是torch 给你用来包装你的数据的工具. 所以你要讲自己的(numpy array 或其他) 数据形式装 ... batch_size=BATCH_SIZE, # mini batch size.
確定! 回上一頁