텐서보드로 파이토치 학습 모니터링 하기




댓글

  1. #============ TensorBoard logging ============#
    # (1) Log the scalar values
    info = {
    'loss': loss.data[0],
    'accuracy': accuracy.data[0]
    }

    for tag, value in info.items():
    logger.scalar_summary(tag, value, step+1)

    # (2) Log values and gradients of the parameters (histogram)
    for tag, value in net.named_parameters():
    tag = tag.replace('.', '/')
    logger.histo_summary(tag, to_np(value), step+1)
    logger.histo_summary(tag+'/grad', to_np(value.grad), step+1)

    # (3) Log the images
    info = {
    'images': to_np(images.view(-1, 28, 28)[:10])
    }

    for tag, images in info.items():
    logger.image_summary(tag, images, step+1)

    답글삭제

댓글 쓰기

이 블로그의 인기 게시물

파이썬으로 Homomorphic Filtering 하기

파이썬으로 2D FFT/iFFT 하기: numpy 버전