Improve an autograd warning (#48765)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/48764
Pull Request resolved: https://github.com/pytorch/pytorch/pull/48765
Reviewed By: heitorschueroff
Differential Revision: D25304145
Pulled By: albanD
fbshipit-source-id: e818413bf92ad0aa382eda77448183b9fd7d5e77
