Skip to content

core.contexts.training.picsellia_training_context

picsellia_training_context

Classes:

Name Description
PicselliaTrainingContext

Context for training jobs in Picsellia, managing parameters and experiment metadata.

PicselliaTrainingContext(hyperparameters_cls, augmentation_parameters_cls, export_parameters_cls, api_token=None, host=None, organization_id=None, organization_name=None, experiment_id=None, working_dir=None)

Bases: PicselliaContext, Generic[THyperParameters, TAugmentationParameters, TExportParameters]

Context for training jobs in Picsellia, managing parameters and experiment metadata.

Methods:

Name Description
to_dict

Convert the context to a dictionary representation.

Attributes:

Name Type Description
experiment_id
experiment
hyperparameters
augmentation_parameters
export_parameters
working_dir str

Return the working directory path for the experiment.

api_token
host
organization_id
organization_name
client

experiment_id = experiment_id or os.getenv('experiment_id') instance-attribute

experiment = self._initialize_experiment() instance-attribute

hyperparameters = hyperparameters_cls(log_data=parameters_log_data) instance-attribute

augmentation_parameters = augmentation_parameters_cls(log_data=parameters_log_data) instance-attribute

export_parameters = export_parameters_cls(log_data=parameters_log_data) instance-attribute

working_dir property

Return the working directory path for the experiment.

api_token = api_token or os.getenv('api_token') instance-attribute

host = host or os.getenv('host', 'https://app.picsellia.com') instance-attribute

organization_id = organization_id or os.getenv('organization_id') instance-attribute

organization_name = organization_name or os.getenv('organization_name') instance-attribute

client = self._initialize_client() instance-attribute

to_dict()

Convert the context to a dictionary representation.