add_graph在Verbose上工作并打印结果,但不保存在日志文件中

时间:2019-08-10 18:18:01

标签: pytorch tensorboard

在新的Pytorch 1.2和add_graph函数的utils中使用tensorboard时,该图不会添加到tensorboard日志文件中。如果我在调用add_graph之前记录了一个标量,它将被记录下来,但是之后添加的任何内容都不会记录。

我将无法在此处提供整个模型,它使用LSTM和Attention,但没有什么花哨的地方。我称之为add_graph的代码段:

    from torch.utils.tensorboard import SummaryWriter

    w = SummaryWriter('./runs/jointmodel')

    dummy_input = torch.Tensor(torch.rand(2, 10, 1024))
    w.add_scalar('log', 10, 1)
    w.add_graph(model, dummy_input, True)
    w.add_scalar('log', 10, 2)

第一个标量被记录,第二个没有记录。 add_graph在控制台中打印整个模型图。生成的文件为空,仅记录了第一个标量。前向通过对模型正常工作。 add_graph的详细信息显示510个元素。任何帮助将不胜感激。

pytorch版本:1.2 Tensorboard版本:1.15.0a20190806

1 个答案:

答案 0 :(得分:0)

显然需要添加冲洗功能。

#include <stdio.h>
#include <stdlib.h>
#include "blepz.h"

int main()
{
    GOO = create_noobza();
    setfnarp(GOO,5);
    int loogaboo = getfnarp(GOO);

    printf("%i", loogaboo);
    return 0;
}

做个好主意。