Hello,
I am working on a similar network similar to shared weight networks:

This image is obtained from the following blog post, in which the author mentioned implementing separate batch normalizations for each network:

  
      
      alexirpan.com
  
  
    

On The Perils of Batch Norm 13

This post is written for deep learning practitioners, and assumes you
know what batch norm is and how it works.


  
  
    
    
  
  


Any ideas on how to implement that in PyTorch please?
Thank you very much in advance for your help!