from torch.utils.data import DataLoader, Dataset ... train_dataloader = DataLoader(train_segment, batch_size=4, shuffle=True, num_workers=0)
確定! 回上一頁