튜토리얼: 데이터셋 아티팩트 생성, 추적 및 사용

import wandb

wandb.login()

# W&B Run을 생성합니다. 이 예제는 데이터셋 아티팩트를 생성하는 방법을 보여주므로
# 작업 유형으로 'dataset'을 지정합니다.
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")

# 아티팩트 버전을 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에서 아티팩트를 쿼리하고 이 run의 입력으로 표시합니다.
  artifact = run.use_artifact("bicycle-dataset:latest")

  # 아티팩트의 내용을 다운로드합니다.
  artifact_dir = artifact.download()