@valhalla thanks for your reply! I have done it before writing on the forum, the author has answered that he has just used the save/load pretrained from your library so the problem might be related to this methods.