메인 콘텐츠로 건너뛰기
W&B Weave를 사용하면 코드의 함수 호출을 트레이스하고, 해당 호출이 실행된 run에 직접 연결할 수 있습니다. @weave.op()으로 함수를 트레이스하고 wandb.init() 컨텍스트 내에서 호출하면 Weave가 해당 트레이스를 W&B run에 자동으로 연결합니다. Traces 테이블에는 연결된 run의 링크가 표시됩니다.

​
Traces 테이블에서 W&B run 보기

다음 Python 코드는 wandb.init() 컨텍스트 내에서 실행할 때 Weave가 트레이스된 Ops를 W&B run에 어떻게 연결하는지 보여줍니다. 이러한 트레이스는 Weave UI에 표시되며 해당 run과 연결됩니다.W&B run을 Weave 트레이스로 보려면 다음과 같이 하세요.
  1. 터미널에서 의존성을 설치합니다.
pip install wandb weave
  1. W&B에 로그인합니다.
wandb login
  1. 다음 스크립트에서 your-team-name/your-project-name을 W&B entity와 프로젝트로 바꾸세요.
import wandb
import weave

def example_wandb(projname):
    # projname을 entity와 project로 분리합니다
    entity, project = projname.split("/", 1)

    # 트레이싱을 위한 Weave 컨텍스트를 초기화합니다
    weave.init(projname)

    # 트레이스 가능한 Op를 정의합니다
    @weave.op()
    def say(message: str) -> str:
        return f"I said: {message}"

    # 첫 번째 W&B run
    with wandb.init(
        entity=entity,
        project=project,
        notes="Experiment 1",
        tags=["baseline", "paper1"],
    ) as run:
        say("Hello, world!")
        say("How are you!")
        run.log({"messages": 2})

    # 두 번째 W&B run
    with wandb.init(
        entity=entity,
        project=project,
        notes="Experiment 2",
        tags=["baseline", "paper1"],
    ) as run:
        say("Hello, world from experiment 2!")
        say("How are you!")
        run.log({"messages": 2})

if __name__ == "__main__":
    # 여기를 실제 W&B username/project로 바꾸세요
    example_wandb("your-team-name/your-project-name")
  1. 스크립트를 실행합니다.
python weave_trace_with_wandb.py
  1. W&B app으로 이동해 프로젝트를 선택합니다.
  2. Weave 프로젝트 sidebar에서 Traces를 클릭합니다. Traces 테이블에 연결된 run의 링크가 표시됩니다.
이제 Traces 테이블에서 트레이스를 열어 연결된 W&B run을 확인하고, 연결된 실험 컨텍스트를 검토할 수 있습니다.