import random
import wandb
PROJECT = "your-project-name"
# ハイパーパラメーターを含む辞書
config = {
'epochs' : 10,
'lr' : 0.01
}
with wandb.init(project=PROJECT, config=config) as run:
offset = random.random() / 5
print(f"lr: {config['lr']}")
# トレーニング run をシミュレーションする
for epoch in range(2, config['epochs']):
acc = 1 - 2**-config['epochs'] - random.random() / config['epochs'] - offset
loss = 2**-config['epochs'] + random.random() / config['epochs'] + offset
print(f"epoch={config['epochs']}, accuracy={acc}, loss={loss}")
run.log({"accuracy": acc, "loss": loss})