def METHOD_NAME(self):
    self.trainer = MMFTrainerMock(self.config_no_oss, 100, 2, 0.04)
    self.trainer.load_datasets()
    self.assertFalse(isinstance(self.trainer.optimizer, OSS))
    self.assertFalse(isinstance(self.trainer.model, ShardedDataParallel))
    self.assertTrue(
        isinstance(self.trainer.model, torch.nn.parallel.DistributedDataParallel)
    )
    self.assertFalse(isinstance(self.trainer.scaler, ShardedGradScaler))
    self.assertTrue(isinstance(self.trainer.scaler, torch.cuda.amp.GradScaler))
    self.assertEqual(self.trainer.current_iteration, 0)
    self.trainer.training_loop()
    self.assertEqual(self.trainer.current_iteration, 4)