airfRANS-model-exploration/configs/remote_wandb_smoke.toml

81 lines
1.7 KiB
TOML
Raw Permalink Normal View History

[run]
name = "airfrans-wandb-smoke"
timeout_minutes = 30
local_artifact_dir = "artifacts/remote_runs"
max_attempts = 1
artifact_sync_interval_seconds = 60
[provider]
kind = "vastai"
disk_gb = 32
max_price_per_hour = 0.09
image = "vastai/base:0.0.2"
[provider.gpu]
name = "RTX 3060 Ti"
count = 1
min_vram_gb = 8
[selection]
min_reliability = 0.90
min_down_mbps = 50
min_up_mbps = 5
require_verified = true
blocked_geos = ["CN"]
blacklist_hosts = [59017, 187467, 155385, 466789]
drop_cheap_frac = 0.0
image_size_gb = 5.0
base_url = "https://cloud.vast.ai"
[workspace]
workdir = "."
exclude = [
"/artifacts",
"/data/raw",
"/.venv",
"/notebooks",
".env",
"HF_TOKEN",
"WANDB_API_KEY",
"__pycache__",
"*.pyc",
]
[bootstrap]
command = """
uv sync --no-dev
uv run --no-dev python -c "import wandb; print('wandb_ok')"
"""
[job]
command = """
uv run --no-dev remote-run wandb-smoke --artifact-dir artifacts/current_run --run-id "$AIRFRANS_REMOTE_RUN_ID" --entity zacheryasc-personal --project airfRANS-model-sweep --hf-repo-url https://huggingface.co/zacheryasc/airfrans-hf-smoke
"""
artifact_dir = "artifacts/current_run"
heartbeat_file = "artifacts/current_run/heartbeat.json"
metrics_file = "artifacts/current_run/metrics.jsonl"
[artifacts]
mode = "rsync"
required = [
"config.toml",
2026-07-28 18:04:50 +00:00
"job_manifest.json",
"metrics.jsonl",
"latest_metrics.json",
"heartbeat.json",
"checkpoint_latest.pt",
"checkpoint_best.pt",
"checkpoint_final.pt",
"final_metrics.json",
"run_manifest.json",
"environment_manifest.json",
"wandb_smoke_manifest.json",
"artifact_manifest.json",
"checksums.txt",
2026-07-28 18:04:50 +00:00
"utilization.jsonl",
]
[cleanup]
on_success = "sky_down"
on_failure = "collect_then_keep"