별칭으로 아티팩트 버전 참조하기

​

기본 별칭

• latest 별칭: 컬렉션에 가장 최근에 연결한 아티팩트 버전에 부여됩니다.
• 고유한 버전 번호: W&B는 연결되는 아티팩트 버전을 0부터 세어, 그 숫자를 바탕으로 각 아티팩트에 고유한 버전 번호를 부여합니다.

​

맞춤형 별칭

• dataset_version_v0, dataset_version_v1, dataset_version_v2 같은 별칭을 사용하여 모델이 어떤 데이터셋으로 트레이닝되었는지 파악할 수 있습니다.
• best_model 별칭을 사용하여 성능이 가장 좋은 아티팩트 모델 버전을 추적할 수 있습니다.

import wandb

# run 초기화
with wandb.init(entity = "<team_entity>", project = "<project_name>") as run:

    # 아티팩트 객체 생성
    # type 매개변수는 아티팩트 객체의 유형과
    # 컬렉션 유형을 모두 지정합니다
    artifact = wandb.Artifact(name = "<name>", type = "<type>")

    # 아티팩트 객체에 파일을 추가합니다.
    # 로컬 머신에 있는 파일의 경로를 지정하세요.
    artifact.add_file(local_path = "<local_path_to_artifact>")

    # 아티팩트를 연결할 컬렉션과 레지스트리 지정
    REGISTRY_NAME = "<registry_name>"
    COLLECTION_NAME = "<collection_name>"
    target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"

    # 아티팩트 버전을 컬렉션에 연결
    # 이 아티팩트 버전에 하나 이상의 별칭 추가
    run.link_artifact(
        artifact = artifact, 
        target_path = target_path, 
        aliases = ["<alias_1>", "<alias_2>"]
        )

​

보호된 별칭

• Production: 프로덕션에서 사용할 준비가 된 아티팩트 버전입니다.
• Staging: 테스트할 준비가 된 아티팩트 버전입니다.

​

보호된 별칭 만들기

1. https://forge.coreweave.com/registry/ 에서 Registry로 이동하세요.
2. 레지스트리를 선택하세요.
3. 페이지 오른쪽 상단의 톱니바퀴 버튼을 클릭하여 레지스트리 설정을 여세요.
4. Protected Aliases 섹션에서 + 버튼을 클릭하여 보호된 별칭을 하나 이상 추가하세요.

1. https://forge.coreweave.com/registry/ 에서 Registry로 이동하세요.
2. 컬렉션에서 View details 버튼을 클릭하세요.
3. Versions 섹션에서 원하는 아티팩트 버전의 View 버튼을 선택하세요.
4. Aliases 필드 옆의 + 버튼을 클릭하여 보호된 별칭을 하나 이상 추가하세요.

​

기존 별칭 찾기

1. https://forge.coreweave.com/registry/ 에서 Registry로 이동하세요.
2. 페이지 상단의 검색 창에 검색어를 입력한 다음 Enter 키를 눌러 검색하세요.

​

예시

import wandb

# run 초기화
with wandb.init(entity = "smle-reg-team-2", project = "zoo_experiment") as run:

    # 아티팩트 객체 생성
    # type 매개변수는 아티팩트 객체의 유형과
    # 컬렉션 유형을 함께 지정합니다
    artifact = wandb.Artifact(name = "zoo_dataset", type = "dataset")

    # 아티팩트 객체에 파일을 추가합니다.
    # 로컬 머신의 파일 경로를 지정하세요.
    artifact.add_file(local_path="zoo_dataset.pt", name="zoo_dataset")
    artifact.add_file(local_path="zoo_labels.pt", name="zoo_labels")

    # 아티팩트를 연결할 컬렉션과 레지스트리를 지정합니다
    REGISTRY_NAME = "Model"
    COLLECTION_NAME = "Zoo_Classifier_Models"
    target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"

    # 아티팩트 버전을 컬렉션에 연결합니다
    # 이 아티팩트 버전에 별칭을 하나 이상 추가합니다
    run.link_artifact(
        artifact = artifact,
        target_path = target_path,
        aliases = ["production-us", "production-eu"]
        )

1. 먼저 아티팩트 객체(wandb.Artifact())를 생성합니다.
2. 다음으로 wandb.Artifact.add_file()을 사용하여 데이터셋 PyTorch 텐서 두 개를 아티팩트 객체에 추가합니다.
3. 마지막으로 link_artifact()를 사용하여 아티팩트 버전을 Zoo_Classifier_Models 레지스트리의 Model 컬렉션에 연결합니다. 이때 aliases 매개변수에 production-us와 production-eu를 인수로 전달하여 아티팩트 버전에 맞춤형 별칭 두 개도 추가합니다.