Current code experiences some issues relaled to loading models that were previously saved. The problem is well known amongst pytorch community and actually pretty common. It is explained here and here (it is more of a pickle module problem as it turns out).
Well, I will rewrite the whole save/load procedure for a better experience. I've chosen an error-prone method for model saving, so we will be better of just changing it completely early in the project.
Current code experiences some issues relaled to loading models that were previously saved. The problem is well known amongst pytorch community and actually pretty common. It is explained here and here (it is more of a pickle module problem as it turns out).