Python 3.6 Pytorch 1.5 Tensorboard Logger CUDA 10.1 Note: Our code works with different PyTorch and CUDA versions, for high version of Pytorch, you need to change one line of our code according to ...