I think you can just re-initialize the Dataset and the DataLoader and change the size in transforms (assuming working with images?), or is there any particular reason you think this is not a good idea?