I’m not sure if the posted shape ([745, 3, 132, 132]) is a single sample or your complete dataset.
In the latter case, you could simply use a TensorDataset, pass it to a DataLoader, and remove the currently used Dataset from the code.