def imagetransformer_imagenet32_base():
  """Best config for ImageNet-32 with 3.77 bits/dim using cross entropy."""
  hparams = imagetransformer_cifar10_base()
  hparams.batch_size = 4
  hparams.layer_prepostprocess_dropout = 0.1
  return hparams
Best config for ImageNet-32 with 3.77 bits/dim using cross entropy.
tensor2tensor/models/image_transformer.py