feat: Limit version of pytorch to 1.11.*

Merged Marko Kollo requested to merge pytorch_version into master

Merge request reports