title: How can I pick a subset of a dataset in Pytorch?
body: <p>I'm trying to run <a href="https://github.com/menardai/FashionGenAttnGAN" rel="nofollow noreferrer">https://github.com/menardai/FashionGenAttnGAN</a> in Google Colab on GPU with the disk size of 30 GB. The code file with its dataset files are about 15 GB. After I extract this code, the disk remains is about 14 GB. When I try to run the Pretrain.py, I can see the captions are loading but suddenly I got the &quot;Assertion Error&quot;. Since I've not got any proper answer for the cause of this error, I think that it is because of my lack of space in Colab environment. The solution that came to my mind is that write some code to tell the model to select just 30% of the train and test datasets to load. But I don't know how to do this. Can anyone help me please?</p>