marwa1:

Are you mean that PyTorch `1.5’ is the reason not an error in my script


Yes, since (some) assert statements were broken in PyTorch 1.5, thus you would have to update to 1.6 or the nightly binary.



 marwa1:

I noticed that the code stoped at the same line in many time :
impf=impf.to(device)


Is this a new issue or why is the code not crashing anymore?