Solved by cbalioglu in post #4 
              
              
                I tested it on a Fedora machine and although it does not get terminated, I does get stuck on aa = self.t(ss) line. 
The problem is that you are calling self.tt.load_state_dict() in the parent process and then use self.tt in the forked process. There are known issues in “implicitly” passing tensors v…