Existem ferramentas para monitorar o treinamento da rede no PyTorch?
Eu estou procurando um equivalente ao tensorboard no tensorflow.
Existem ferramentas para monitorar o treinamento da rede no PyTorch?
Eu estou procurando um equivalente ao tensorboard no tensorflow.
Respostas:
Estou usando o tensorboardX . Ele suporta a maioria dos recursos do TensorBoard. Estou usando o escalar, imagens, distribuições, histogramas e texto. Ainda não tentei o resto como áudio e gráfico. Mas o repo também contém exemplos para esses casos. A instalação pode ser feita facilmente com o pip. Está tudo explicado no leia-me.
Também existem outros softwares que implementam um wrapper para o PyTorch (e outras linguagens / estruturas) do TensorBoard. Tanto quanto eu sei, eles suportam menos funcionalidades. Mas dê uma olhada no Crayon e no Tensorboard-Logger .
Também estou usando o Tensorboard-PyTorch (TensorboardX). Acima de tudo, estou muito feliz com isso.
Mas não tente visualizar gráficos. Pelo menos nenhum com um pouco de complexidade (por exemplo, um resnet50 não funcionará). Existem alguns problemas sobre isso na página do github .
Mas é melhor consultar o fórum Pytorch com frequência. Pytorch parece se mover muito rápido. E também está planejada uma implementação direta do tensorboard .
Sim, existem algumas ferramentas: TNT, tensorboardX e VisualDL. Dê uma olhada nesta postagem do blog https://nbro.github.io/blogging/2019/01/06/an-example-of-how-to-use-visualdl-with-pytorch/ , onde um exemplo de como use VisualDL é fornecido, para mais detalhes.