rlworkgroup / dowel

A little logger for machine learning research
MIT License
32 stars 37 forks source link

TensorBoardOutput's resource is not cleaned up nicely #28

Open zhanpenghe opened 5 years ago

zhanpenghe commented 5 years ago

When using the logger with tensorboard, it will hang if we don't call logger.remove_all(). Here is a simple snippet for reproduce it:

import dowel
from dowel import logger

logger.add_output(dowel.TensorBoardOutput(log_dir))

This is mostly due to tensorboardX.SummaryWriter failed to clean up its resource.