code: def setUp(self):
      super().setUp()
      # Make sure tf does not allocate gpu memory.
      tf.config.experimental.set_visible_devices([], "GPU")
  
file_path: flax/examples/imagenet/train_test.py
imports: import pathlib
  import tempfile
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import random
  import tensorflow
  import tensorflow_datasets
  import models
  import train
  from configs import default
label_desc: Use image node_id 1 for calling the TrainTest obj's underlying member method code with example usage: obj.setUp() without return types