code: def on_state_reset():
      for param_group in optimizer.param_groups:
          param_group["lr"] = args.lr * hvd.size()
  
file_path: horovod/examples/elastic/pytorch/pytorch_mnist_elastic.py
imports: import argparse
  import os
  from filelock import FileLock
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from torchvision import datasets, transforms
  import torch.utils.data.distributed
  import horovod.torch
label_desc: Use image node_id 4 for calling a global function with example usage: on_state_reset() without return types