如何在运行结束前从wandb中检索已记录的值?
import osimport wandbwandb.init(project='someproject')def loss_a(): # do_stuff and log: wandb.log({"loss_a": 1.0}) def loss_b(): # do_stuff and log: wandb.log({"loss_b": 2.0})for epoch in range(2): loss_a() loss_b() # 以某种方式检索loss_a和loss_b并在这里打印它们: print(f'loss_a={??}, loss_b={??}')
在运行结束后,我可以使用wandb.Api
来查找run.history
。但似乎在运行结束前,访问run.history
不起作用。
回答:
您可以在运行结束前通过使用wandb.run.summary
来检索wandb中已记录的值。这是一个字典,保存了特定键名的最后记录值。请查看此链接。
您还可以查看这个colab笔记本来自己尝试一下。