Advertisement

mxnet使用mxboard可视化模型权重参数报错:No handlers could be found for logger “mxboard.event_file_writer“

阅读量:

在日志记录过程中,未能找到适用于“mxboard.event_file_writer”这一模块的处理程序。

处理方式:

复制代码
    $ pip install tensorboard
    
    
    bash

在代码中添加以下内容:

复制代码
    import logging
    logging.basicConfig(level=logging.DEBUG)
    
    
    python

https://github.com/reminisce/mxboard- demo /blob/master/train_mnist.py

训练过程中可视化展示的实现

复制代码
    # visualize
    from mxboard import SummaryWriter
    import logging
    logging.basicConfig(level=logging.DEBUG)
    sw = SummaryWriter(logdir='./logs', flush_secs=5)
    net = mfc1
    argp,auxp = net.get_params()
    grads = [i for i in argp.values()]
    
    param_name

全部评论 (0)

还没有任何评论哟~