Correct pyre errors for trainer
Correct pyre errors for trainer

Summary: Correct pyre errors for trainer

Reviewed By: patricklabatut

Differential Revision: D25974873

fbshipit-source-id: a69d2c8f83dff2cceb01004b2b6aeb5fe655d4e8

facebookresearch/detectron2,facebookresearch/detectron2,facebookresearch/detectron2