PyTorch + Skorch:使TensorBoard callbakc跟踪由`EpochScoring`定义的手动分数

kqhtkvqz  于 2023-03-23  发布在  其他
关注(0)|答案(1)|浏览(132)

我用的是Skorch
我使用EpochScoring回调创建了一个新的分数记录。
虽然它出现在历史日志中,但它不会显示在TensorBoard上(我使用的是TensorBoard回调)。
有没有办法让TensorBoard记录器知道该键?
我知道我可以做一个手动回调,但我想知道是否有一个简单的方法来让记录器从历史中保存一个现有的密钥。

ws51t4hk

ws51t4hk1#

在这种情况下,调用顺序可能很重要。因为两个回调都在epoch结束时工作,所以TensorBoard回调只能访问执行时写入的历史记录中的那些键。
因此,网络的回调列表应该如下所示:

net = NeuralNet(
   ...,
   callbacks=[
       EpochScoring(...),
       TensorBoard(...),
   ],
)

如果是另一种情况,新的乐谱就不会写出来了。

相关问题