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