qualia_core.experimenttracking.pytorch package

Submodules

• qualia_core.experimenttracking.pytorch.ClearML module
  • ClearML
    • ClearML.start()
    • ClearML.stop()
    • ClearML.initializer()
    • ClearML.import_and_clear_all_offline_sessions()
    • ClearML.logger
• qualia_core.experimenttracking.pytorch.ExperimentTrackingPyTorch module
  • ExperimentTrackingPyTorch
    • ExperimentTrackingPyTorch.logger
• qualia_core.experimenttracking.pytorch.Neptune module
  • Neptune
    • Neptune.start()
    • Neptune.stop()
    • Neptune.logger

Module contents

class qualia_core.experimenttracking.pytorch.ClearML(project_name: str, task_name: str, sources_path: str | None = None, ignores: list[str] | None = None, offline_mode: bool = False)

Bases: ExperimentTracking

start(name: str | None = None) → None

stop() → None

classmethod initializer() → None

Connect to task in PyTorch Lightning Trainer subprocess with e.g. ddp_spawn.

classmethod import_and_clear_all_offline_sessions() → None

property logger: None

class qualia_core.experimenttracking.pytorch.Neptune(project_name: str, config_file: str = 'conf/neptune.toml')

Bases: NeptuneBase, ExperimentTrackingPyTorch

start(name: str | None = None) → None

stop() → None

property logger: NeptuneLogger