AS2 / src /softasp /cli /train_scratch.py

Release AS2 framework with Hydra-selected instantiations

69928d6 verified 9 days ago

703 Bytes

"""Hydra entry point for training an AS2 instantiation without supplied weights.
Run with ``uv run --locked softasp-train-scratch``.
"""

import hydra
from omegaconf import DictConfig
from softasp_core.utils.logging import setup_logging

from softasp.configuration import config_directory
from softasp.instantiations.runtime import train_from_scratch

_CONFIGS = config_directory()


@hydra.main(version_base="1.3", config_path=_CONFIGS, config_name="train_scratch")
def main(cfg: DictConfig) -> None:
    setup_logging(level=cfg.logging.get("level", "INFO"))
    manifest = train_from_scratch(cfg)
    print(f"Training workflow complete. Manifest: {manifest}")


if __name__ == "__main__":
    main()

	"""Hydra entry point for training an AS2 instantiation without supplied weights.

	Run with ``uv run --locked softasp-train-scratch``.
	"""

	import hydra
	from omegaconf import DictConfig
	from softasp_core.utils.logging import setup_logging

	from softasp.configuration import config_directory
	from softasp.instantiations.runtime import train_from_scratch

	_CONFIGS = config_directory()


	@hydra.main(version_base="1.3", config_path=_CONFIGS, config_name="train_scratch")
	def main(cfg: DictConfig) -> None:
	setup_logging(level=cfg.logging.get("level", "INFO"))
	manifest = train_from_scratch(cfg)
	print(f"Training workflow complete. Manifest: {manifest}")


	if __name__ == "__main__":
	main()