Launch エージェント

ソースコードを表示

指定された run queue をポーリングし、 wandb launch の Runs を起動する Launch agent クラスです。

LaunchAgent(
    api: Api,
    config: Dict[str, Any]
)

引数	説明
api	バックエンドへのリクエストに使用する Api オブジェクト。
config	エージェントのコンフィグ 辞書。

属性	説明
num_running_jobs	スケジューラーを含まない、実行中のジョブ数を返します。
num_running_schedulers	スケジューラーの数のみを返します。
thread_ids	エージェントで実行中のスレッド ID の キー リストを返します。

​

メソッド

​

check_sweep_state

ソースを表示

check_sweep_state(
    launch_spec, api
)

Sweep の Run を起動する前に、 Sweeps の状態を確認します。

​

fail_run_queue_item

ソースを表示

fail_run_queue_item(
    run_queue_item_id, message, phase, files=None
)

​

finish_thread_id

ソースを表示

finish_thread_id(
    thread_id, exception=None
)

現在のリストからジョブを削除します。

​

get_job_and_queue

ソースを表示

get_job_and_queue()

​

initialized

ソースを表示

@classmethod
initialized() -> bool

エージェントが初期化されているかどうかを返します。

​

loop

ソースを表示

loop()

ジョブのポーリングと実行を無限ループで行います。

例外	説明
KeyboardInterrupt	エージェントの停止が要求された場合に発生します。

​

name

ソースを表示

@classmethod
name() -> str

エージェントの名前を返します。

​

pop_from_queue

ソースを表示

pop_from_queue(
    queue
)

ジョブとして実行するために、 runqueue からアイテムを取り出します。

引数	説明
queue	取り出し対象のキュー。

戻り値	説明
キューから取り出されたアイテム。

例外	説明
Exception	キューからの取り出し中にエラーが発生した場合。

​

print_status

ソースを表示

print_status() -> None

エージェントの現在のステータスを表示します。

​

run_job

ソースを表示

run_job(
    job, queue, file_saver
)

Projects を設定し、ジョブを実行します。

引数	説明
job	実行するジョブ。

​

task_run_job

ソースを表示

task_run_job(
    launch_spec, job, default_config, api, job_tracker
)

​

update_status

ソースを表示

update_status(
    status
)

エージェントのステータスを更新します。

引数	説明
status	更新後のエージェントのステータス。