YU
chainer_extensions
Installation
- Install chainer
- Install tensorboardX
pip install chainer
pip install tensorboardXParameterStatisticsX
from extensions import ParameterStatisticsX
trainer.extend(ParameterStatisticsX(model, prefix='model'))
tensorboard --logdir=./result/.tensorboardOutputting images by the dot command
from extensions import graphviz_dot
trainer.extend(graphviz_dot(file_name='*.dot'))