Hi,
My variables were already on volatile.
I’m gonna try to squeeze the sequential, which is used only to simplify the code.
The time spend is a problem too, but it’s related to this issue on github :  https://github.com/pytorch/pytorch/issues/4703 19
My first problem is still the same. Whene using twice the same data, the same network architecture but two different checkpoints, my runtime change a lot.
I’m trying different changes related to the limits of my CPU, like using cropped images or smaller network. The problem is still the same for now