Hi,
Recently, I have started to work on larger images (brain), and I found a big slow down going from size (1, 2, 224, 224, 224) to (1, 2, 256, 256, 256) I was expecting something slower but not by this amount:

(1, 2, 224, 224, 224): torch::autograd::CopyBackwards 402133.723us 0.000us 2 804267.446us 0.000us

(1, 2, 256, 256, 256): torch::autograd::CopyBackwards 10681252.066us 0.000us 2 21362504.132us 0.000us


Here is the code 2 with the output of the torch.autogred.profiler.profile().
I tested it on pytorch 0.4.1 and 1.0.0 and I have the same behavior.
I don’t know what I’m doing wrong with pytorch, someone have any idea?
Thanks