Renamed deit experiment

34c6530a · Simon Geisler · a2800201 · 34c6530a
Commit 34c6530a authored Aug 19, 2021 by Simon Geisler
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/projects/vit/configs/image_classification.py .../vision/beta/projects/vit/configs/image_classification.py +1 -1

No files found.
--- a/official/vision/beta/projects/vit/configs/image_classification.py
+++ b/official/vision/beta/projects/vit/configs/image_classification.py
@@ -81,7 +81,7 @@ task_factory.register_task_cls(ImageClassificationTask)(
    image_classification.ImageClassificationTask)


-@exp_factory.register_config_factory('deit_imagenet_pretrain_nodistillation')
+@exp_factory.register_config_factory('deit_imagenet_pretrain')
 def image_classification_imagenet_deit_pretrain() -> cfg.ExperimentConfig:
  """Image classification on imagenet with vision transformer."""
  train_batch_size = 4096  # originally was 1024 but 4096 better for tpu v3-32