torch>=1.8.0
tensorboard>=2.4.0
