Update import of SummaryWriter to use one from torch.

Author: BloodAxe

pytorch_toolbelt/utils/catalyst/visualization.py

@@ -9,7 +9,8 @@
 from torch import Tensor
 from catalyst.dl import Callback, CallbackOrder, IRunner, CallbackNode
 from catalyst.dl.callbacks import TensorboardLogger
-from catalyst.contrib.tools.tensorboard import SummaryWriter
+from torch.utils.tensorboard import SummaryWriter
+
 from pytorch_toolbelt.utils import render_figure_to_tensor
 from pytorch_toolbelt.utils.distributed import all_gather