type hints for pytorch models (#17064)

Commit

3 years ago

type hints for pytorch models (#17064) * type hints for pytorch models * fixed import error * fixed some errors