Artifact 내려받아 사용하기
Artifact 내려받아 사용하기
만들어 둔 Artifact는 다른 Run·다른 사람이 내려받아 그대로 재사용할 수 있어요. use_artifact로 참조만 잡으면 로컬에 파일을 내려받아 원하는 경로에서 쓸 수 있죠.
출처: https://docs.wandb.ai/models/artifacts/download-and-use-an-artifact
기본 사용법은 이렇게 돼요. 프로젝트에서 특정 Artifact 버전을 가져온 뒤, artifact.download()로 파일을 로컬에 받아 Path를 얻고, 그 경로로 데이터셋이나 모델을 로드하면 돼요.
import wandb
run = wandb.init(project="my-project")
artifact = run.use_artifact("my-dataset:v0")
dataset_dir = artifact.download()
print(dataset_dir)
name:alias 또는 name:version 형태로 원하는 버전을 지목해요. "my-dataset:v0"는 v0 버전을, "my-dataset:latest" 같은 별칭(alias)으로도 가리킬 수 있어요. 별칭은 '지금 쓰는 최신 것'처럼 흐름이 자주 바뀌는 대상을 가리킬 때 편리해요.
내려받은 뒤에는 dataset_dir에서 파일을 읽거나, 모델을 torch.load 같은 방식으로 불러오면 됩니다. W&B는 use_artifact로 참조를 선언할 때 부모 Artifact와의 의존 관계도 함께 기록해요. 그래서 이 Run이 어떤 데이터 버전에 의존했는지 나중에 그래프로 다시 확인할 수 있어요.