RuntimeError: CUDA out of memory. Tried to allocate…but memmory is empty

Question

I\'m trying to run the train file from this Unet with their default hyperparameters, batch size = 1.

I have a GTX970 with 4GB and made Windows use the integrated grap