refactor: tensorboard on pytorch

Created by: IgnatovFedor

What problem are we trying to solve?:

Tensorboard used by tensorflow 1.15, that is going be removed.

How can we solve it?:

Rewrite tensorboard usage using pytorch api