code: def on_state_reset():
      optimizer.lr.assign(lr * hvd.size())
  
file_path: horovod/test/integration/data/elastic_tensorflow2_main.py
imports: import argparse
  import json
  import os
  import psutil
  import time
  import tensorflow
  import horovod.tensorflow
label_desc: Use image node_id 5 for calling a global function with example usage: on_state_reset() without return types