Spaces:
Sleeping
Sleeping
| # ------------------------------------------------------------------- | |
| # Pimcore | |
| # | |
| # This source file is available under two different licenses: | |
| # - GNU General Public License version 3 (GPLv3) | |
| # - Pimcore Commercial License (PCL) | |
| # Full copyright and license information is available in | |
| # LICENSE.md which is distributed with this source code. | |
| # | |
| # @copyright Copyright (c) Pimcore GmbH (http://www.pimcore.org) | |
| # @license http://www.pimcore.org/license GPLv3 and PCL | |
| # ------------------------------------------------------------------- | |
| from pydantic import BaseModel | |
| from typing import Annotated | |
| from fastapi import Form | |
| class ImageClassificationTrainingParameters(BaseModel): | |
| """ Provides specific training parameters for the image classification fine tuning.""" | |
| epochs: int | |
| learning_rate: float | |
| def map_image_classification_training_parameters( | |
| epocs: Annotated[int, Form(description="Epochs executed during training.")] = 3, | |
| learning_rate: Annotated[float, Form(description="Learning rate for training.")] = 5e-5 | |
| ) -> ImageClassificationTrainingParameters: | |
| """ Maps the parameters to the ImageClassificationTrainingParameters class. """ | |
| return ImageClassificationTrainingParameters( | |
| epochs=epocs, | |
| learning_rate=learning_rate | |
| ) | |
| class ImageClassificationParameters: | |
| """ Provides all parameters for the image classification fine tuning. """ | |
| __training_files_path: str | |
| __training_zip_file_path: str | |
| __project_name: str | |
| __source_model_name: str | |
| __training_parameters: ImageClassificationTrainingParameters | |
| def __init__(self, | |
| training_files_path: str, | |
| training_zip_file_path: str, | |
| project_name: str, | |
| source_model_name: str, | |
| training_parameters: ImageClassificationTrainingParameters | |
| ): | |
| self.__training_files_path = training_files_path | |
| self.__training_zip_file_path = training_zip_file_path | |
| self.__project_name = project_name | |
| self.__source_model_name = source_model_name | |
| self.__training_parameters = training_parameters | |
| def get_training_files_path(self) -> str: | |
| return self.__training_files_path | |
| def get_training_zip_file(self) -> str: | |
| return self.__training_zip_file_path | |
| def get_result_model_name(self) -> str: | |
| return self.__project_name | |
| def get_project_name(self) -> str: | |
| return self.__project_name | |
| def get_source_model_name(self) -> str: | |
| return self.__source_model_name | |
| def get_training_parameters(self) -> ImageClassificationTrainingParameters: | |
| return self.__training_parameters | |