enable AMP trainer
Summary: similar to (https://github.com/facebookresearch/detectron2/commit/6be4c339f30238a401d198db96ecd50ac169f137)
@@ -12,7 +12,7 @@ from torch.nn.parallel import DataParallel, DistributedDataParallel
import detectron2.utils.comm as comm
from detectron2.utils.events import EventStorage, get_event_storage
-__all__ = ["HookBase", "TrainerBase", "SimpleTrainer"]
+__all__ = ["HookBase", "TrainerBase", "SimpleTrainer", "AMPTrainer"]
class HookBase:
