Skip to content

GitLab

Commit 34c6530a authored by Simon Geisler's avatar Simon Geisler
Browse files

Renamed deit experiment

parent a2800201
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/vit/configs/image_classification.py
View file @ 34c6530a
...@@ -81,7 +81,7 @@ task_factory.register_task_cls(ImageClassificationTask)( ...@@ -81,7 +81,7 @@ task_factory.register_task_cls(ImageClassificationTask)(
image_classification.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: def image_classification_imagenet_deit_pretrain() -> cfg.ExperimentConfig:
"""Image classification on imagenet with vision transformer.""" """Image classification on imagenet with vision transformer."""
train_batch_size = 4096 # originally was 1024 but 4096 better for tpu v3-32 train_batch_size = 4096 # originally was 1024 but 4096 better for tpu v3-32
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment