Yes, I faced the issue too, so basically although I don’t know the reason, but you cannot execute the code a second time without restarting the session (with the same GPU on colab). For me, all my results are reproducible once I make sure they are on the same GPU + restart session.