调用DataLoader
时参数出错
self.train_dataloader = DataLoader(train_dataset, batch_size=TrainOption.train_batch_size,
shuffle=TRUE,num_workers=TrainOption.data_load_worker_num)
shuffle
参数设置错误。
因为已经有batch_sample
了,就不需要shuffle
来进行随机的sample
了,所以在这里的shuffle
应该设置为FALSE
才对。