Native TensorBoard support

Author: gemenerik

main.py

@@ -14,7 +14,7 @@
 import datasets
 from multiscaleloss import multiscaleEPE, realEPE
 import datetime
-from tensorboardX import SummaryWriter
+from torch.utils.tensorboard import SummaryWriter
 from util import flow2rgb, AverageMeter, save_checkpoint
 
 model_names = sorted(name for name in models.__dict__

requirements.txt

@@ -2,7 +2,7 @@ torch>=1.2
 torchvision
 numpy
 spatial-correlation-sampler>=0.2.1
-tensorboardX>=1.4
+tensorboard
 imageio
 argparse
 path