code: def test_lr_schedule(self):
      with patch(
          "mmf.trainers.lightning_trainer.get_mmf_env", return_value=""
      ):
          # note, be aware some of the logic also is in the SimpleLightningModel
          config = self._get_config(max_steps=8, lr_scheduler=True)
          trainer1 = get_lightning_trainer(config=config)
          trainer1.trainer.fit(
              trainer1.model, trainer1.data_module.train_loader
          )
  
          config = self._get_config(max_steps=8)
          trainer2 = get_lightning_trainer(config=config)
          trainer2.trainer.fit(
              trainer2.model, trainer2.data_module.train_loader
          )
  
          last_model_param1 = list(trainer1.model.parameters())[-1]
          last_model_param2 = list(trainer2.model.parameters())[-1]
          self.assertFalse(
              torch.allclose(last_model_param1, last_model_param2)
          )
  
file_path: mmf/tests/trainers/lightning/test_lr_schedule.py
imports: import unittest
  from unittest.mock import MagicMock, patch
  import torch
  from mmf.trainers.callbacks.lr_scheduler import LRSchedulerCallback
  from tests.trainers.test_utils import get_config_with_defaults, get_lightning_trainer, get_mmf_trainer
label_desc: Use image node_id 1 for calling the TestLightningTrainerLRSchedule obj's underlying member method code with example usage: obj.test_lr_schedule() without return types