Different training results on different machines | With simplified test code

[Hassan_Imani (Hsn Iman)]

I created this:

Problem with converting keras code to pytorch code

I dont know how to initialize he_uniform in pytorch. I did not find anything! Could you please help to do that?