This PR changes the current usage from tensorboardX to tensorboard.
Since PyTorch natively supports tensorboard it would be nice to use it instead of tensorboardX.
Added Features:
Added log_tensorboard_figures functionality to add matplotlib figures to the logging mechanism.
This PR changes the current usage from tensorboardX to tensorboard. Since PyTorch natively supports tensorboard it would be nice to use it instead of tensorboardX.
Added Features: Added log_tensorboard_figures functionality to add matplotlib figures to the logging mechanism.