pfd.superop package

Submodules

pfd.superop.exploration_dist_loop module

class pfd.superop.exploration_dist_loop.ExplDistBlock(name: str, prep_run_explore_op: OPTemplate, prep_run_train_op: OPTemplate, collect_data_op: Type[OP], inference_op: Type[OP], inference_config: dict, model_test_config: dict, collect_data_config: dict, upload_python_packages: List[PathLike] | None = None)[source]

Bases: Steps

property input_artifacts
property input_parameters
property output_artifacts
property output_parameters
class pfd.superop.exploration_dist_loop.ExplDistLoop(name: str, expl_dist_blk_op: OPTemplate, scheduler_config: dict, upload_python_packages: List[PathLike] | None = None)[source]

Bases: Steps

property input_artifacts
property input_parameters
property output_artifacts
property output_parameters

pfd.superop.exploration_finetune_loop module

class pfd.superop.exploration_finetune_loop.ExplFinetuneBlock(name: str, prep_run_explore_op: OPTemplate, prep_run_fp_op: OPTemplate, collect_data_op: Type[OP], select_confs_op: Type[OP], prep_run_train_op: OPTemplate, inference_op: Type[OP], collect_data_step_config: dict, inference_step_config: dict, select_confs_step_config: dict, upload_python_packages: List[PathLike] | None = None)[source]

Bases: Steps

property input_artifacts
property input_parameters
property output_artifacts
property output_parameters
class pfd.superop.exploration_finetune_loop.ExplFinetuneLoop(name: str, expl_ft_blk_op: OPTemplate, scheduler_config: dict, upload_python_packages: List[PathLike] | None = None)[source]

Bases: Steps

property input_artifacts
property input_parameters
property output_artifacts
property output_parameters

Module contents