class_docstr: Benchmarks for the ImageNet Flax example.
class_docstr_tok: ["Benchmarks","for","the","ImageNet","Flax","example","."]
docstr: Run ImageNet on 8x V100 GPUs in half precision for 2 epochs.
docstr_tok: ["Run","ImageNet","on","8x","V100","GPUs","in","half","precision","for","2","epochs","."]
code: def test_8x_v100_half_precision_short(self):
      """Run ImageNet on 8x V100 GPUs in half precision for 2 epochs."""
      self._test_8x_v100_half_precision(
          num_epochs=2, min_accuracy=0.06, max_accuracy=0.09
      )
      self.report_extras(
          {
              "description": "Short (2 epochs) 8 x V100 test for ImageNet ResNet50.",
              "model_name": "resnet50",
              "parameters": "hp=true,bs=2048,num_epochs=2",
              "implementation": "linen",
          }
      )
  
file_path: flax/examples/imagenet/imagenet_benchmark.py
imports: import time
  from absl import flags
  from absl.testing import absltest
  from absl.testing.flagsaver import flagsaver
  from flax.testing import Benchmark
  import jax
  import numpy
  import main
  from configs import v100_x8_mixed_precision
label_desc: Use image node_id 2 for calling the ImagenetBenchmark obj's underlying member method code with example usage: obj.test_8x_v100_half_precision_short() without return types