[zero-3] print warning once and support torch parameter (#2127)
* print warning only once.
* add support for torch param and only warn on gpu 0
* remove type checking. will be done on a new PR with more tests.
Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>