code: def main(argv):
      del argv
      print("JAX host: %d / %d" % (jax.host_id(), jax.host_count()))
      print(
          "JAX devices:\n%s" % "\n".join(str(d) for d in jax.devices()),
          flush=True,
      )
      experiment = Experiment()
      experiment.train_and_eval()
      objax.util.multi_host_barrier()
  
file_path: objax/examples/image_classification/imagenet_resnet50_train.py
imports: import os
  import time
  import jax
  import jax.numpy
  import numpy
  from absl import app, flags, logging
  import objax
  from examples.image_classification import imagenet_resnet50_data
  from objax.zoo.resnet_v2 import ResNet50
label_desc: Use image node_id 1 for calling a global function with example usage: main(argv) without return types