| # Copyright (c) Facebook, Inc. and its affiliates. | |
| # | |
| # This source code is licensed under the MIT license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| from .image_pretraining import ImagePretrainingTask, ImagePretrainingConfig | |
| from .image_classification import ImageClassificationTask, ImageClassificationConfig | |
| from .mae_image_pretraining import MaeImagePretrainingTask, MaeImagePretrainingConfig | |
| __all__ = [ | |
| "ImageClassificationTask", | |
| "ImageClassificationConfig", | |
| "ImagePretrainingTask", | |
| "ImagePretrainingConfig", | |
| "MaeImagePretrainingTask", | |
| "MaeImagePretrainingConfig", | |
| ] |