modifications to data reader and problem_hparams

PiperOrigin-RevId: 161027111

Author: Niki Parmar

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='tensor2tensor',
-    version='1.0.10',
+    version='1.0.9',
     description='Tensor2Tensor',
     author='Google Inc.',
     author_email='[email protected]',

tensor2tensor/data_generators/problem_hparams.py

@@ -664,8 +664,17 @@ def image_mscoco_tokens(model_hparams, vocab_count):
   }
   p.batch_size_multiplier = 256
   p.max_expected_batch_size_per_shard = 2
+
+
+def img2img_imagenet(unused_model_hparams):
+  """Image 2 Image for imagenet dataset."""
+  p = default_problem_hparams()
+  p.input_modality = {"inputs": ("image:identity", None)}
+  p.target_modality = ("image:identity", None)
+  p.batch_size_multiplier = 256
+  p.max_expected_batch_size_per_shard = 4
   p.input_space_id = 1
-  p.target_space_id = 3
+  p.target_space_id = 1
   return p
 
 
@@ -732,4 +741,5 @@ def image_mscoco_tokens(model_hparams, vocab_count):
     "image_mscoco_tokens_128k_tune": lambda p: image_mscoco_tokens(p, 2**17),
     "image_mscoco_tokens_128k_test": lambda p: image_mscoco_tokens(p, 2**17),
     "image_imagenet": image_imagenet,
+    "img2img_imagenet": img2img_imagenet,
 }