torch
tensorboard<=2.4.1,>=2.3.0
six
werkzeug
numpy
