Load model pytorch

[SkafteNicki]

Before submitting

• Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
• Did you write any new necessary tests?

What does this PR do?

Fixes #120

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in GitHub issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[Borda]

mind checking:

>           raise TypeError(f"Expected state_dict to be dict-like, got {type(state_dict)}.")
E           TypeError: Expected state_dict to be dict-like, got <class 'torch.nn.modules.module.Module'>.