hello,
Thanks for your reply!
Could you show me an classcifacation tast example which use “torch.utils.data.random_split”, I think the problem is not caused by GPU memory, because when I manually splt the dataset to training and testing dataset, the code works, but I really can access the dataset split by “torch.utils.data.random_split” now, it just show the error when the code start training