sauravkdeo's picture
chore: log more info to tensorboard (#1608)
8a29310