チュートリアル: データセットArtifactを作成、トラッキング、使用する

import wandb

wandb.login()

# W&B の run を作成します。ここでは、このサンプルがデータセット artifact の作成方法を示すため、
# ジョブタイプとして 'dataset' を指定しています。
with wandb.init(project="artifacts-example", job_type="upload-dataset") as run:
    # ここにコードを記述します

artifact = wandb.Artifact(name="bicycle-dataset", type="dataset")

# artifactのコンテンツにファイルを追加する
artifact.add_file(local_path="dataset.h5")

# artifact のバージョンを W&B に保存し、
# この run の出力としてマークする
run.log_artifact(artifact)

import wandb

wandb.login()

with wandb.init(project="artifacts-example", job_type="upload-dataset") as run:
    artifact = wandb.Artifact(name="bicycle-dataset", type="dataset")
    artifact.add_file(local_path="dataset.h5")
    run.log_artifact(artifact)

# W&B の Run を作成します。ここでは 'type' に 'training' を指定します
# この run をトレーニングのトラッキングに使用するためです。
with wandb.init(project="artifacts-example", job_type="training") as run:

  # W&B に artifact をクエリし、この run への入力としてマークします
  artifact = run.use_artifact("bicycle-dataset:latest")

  # artifact の内容をダウンロードします
  artifact_dir = artifact.download()