code: def __init__(
      self,
      context: PyTorchTrialContext,
      hparams: typing.Optional[typing.Dict],
  ) -> None:
      super().__init__(context=context, hparams=hparams)
      self.model, self.optimizer = self.context.configure_apex_amp(
          models=self.model, optimizers=self.optimizer
      )
  
imports: import typing
  from train import MNistTrial
  from determined.pytorch import PyTorchTrialContext
label_desc: Use image node_id 1 to create a new MNistApexAMPTrial object from inherited base classes: MNistTrial with example: obj = MNistApexAMPTrial(context, hparams)