I created this:
  
    
    
    Problem with converting keras code to pytorch code vision
  
  
    I dont know how to initialize he_uniform in pytorch. 
I did not find anything! 
Could you please help to do that?