Added validation of mode parameter in AveragedModel (#65921)
Summary:
Discussion: https://github.com/pytorch/pytorch/pull/65495#issuecomment-930460469
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65921
Reviewed By: albanD
Differential Revision: D31310105
Pulled By: prabhat00155
fbshipit-source-id: 417691832a7c793744830c11e0ce53e3972d21a3