Using TensorBoard for logging and visualizing a PyTorch RL agent's progress