pytorchts==0.2.0
numpy
torch>=1.5.0
tensorboard
