레지스트리 개요

import wandb
import random

# 아티팩트를 추적할 W&B run을 초기화합니다
with wandb.init(project="registry_quickstart") as run:
    # 로깅할 수 있도록 모의 모델 파일을 생성합니다
    with open("my_model.txt", "w") as f:
        f.write("Model: " + str(random.random()))

    # 아티팩트를 W&B에 로깅합니다
    logged_artifact = run.log_artifact(
        artifact_or_path="./my_model.txt", 
        name="mnist_model", 
        type="model" # 아티팩트 유형을 지정합니다
    )

    # 아티팩트를 게시할 컬렉션과 레지스트리의
    # 이름을 지정합니다
    COLLECTION_NAME = "first-collection"
    REGISTRY_NAME = "model"

    # 아티팩트를 레지스트리에 연결합니다
    run.link_artifact(
        artifact=logged_artifact, 
        target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"
    )