memory issues

[dnettlet]

Hi,
 
We are having some issues running "Pytorch_CamouflageImages".

We tried in local but ran out of memory on our desktop:

RuntimeError: [enforce fail at ..\c10\core\CPUAllocator.cpp:76] data. DefaultCPUAllocator: not enough memory: you tried to allocate 67108864 bytes.

So we tried in "Google Colab" but it exits at epoch 77, and the images written at epochs 0, 30 and 60 are all the same. Maybe does not start to change until later (closer to 1000 epochs)?

Maybe in google colab we need to set some memory allocation?

Thanks in advance and best regards,
David N.