@ptrblck
Any update on this?
I also tested the script on another Linux machine and the result is the same.
When GPU is enabled, it would produce this weird behavior.
And interestingly, it only happens when the network is involved.
I can run the code without the network (i.e. remove line 134~144) and the result is fine on GPU.
I also tried creating a new tensor instead of using repeat function and it had the same issue.
Additionally, I received an error on Linux when trying to save the images.
It says “ValueError: ndarray is not C-contiguous”. However, this error is not presented on Windows machine.
Could the contiguous property be the cause because of the reshape function?
Thanks.