import wandb
with wandb.init(project="my-models") as run:
# モデルをトレーニングします
model = train_model()
# モデルの Artifact を作成します
model_artifact = wandb.Artifact(
name="my-model",
type="model",
description="ResNet-50 trained on ImageNet subset",
metadata={
"architecture": "ResNet-50",
"dataset": "ImageNet-1K",
"accuracy": 0.95
}
)
# Artifact にモデルファイルを追加します
model_artifact.add_file("model.pt")
model_artifact.add_dir("model_configs/")
# Artifact を W&B にログ記録します
run.log_artifact(model_artifact)