Solved by Prashant in post #4 
              
              
                Looks like inheritance is not supported Unable to call `super` method with TorchScript · Issue #42885 · pytorch/pytorch · GitHub