Alguma ferramenta pytorch para monitorar o treinamento da rede neural?

7

Existem ferramentas para monitorar o treinamento da rede no PyTorch?

Eu estou procurando um equivalente ao tensorboard no tensorflow.

Chati Denati
fonte

Respostas:

4

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 .

Fritz O'Connor
fonte
2

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 .

SeparateReality
fonte