Solved by albanD in post #2 
              
              
                Hi, 
This is linked to when the load_state_dict() function is called. 
Note that this function is internal and might change without notice in the future.