torchvision==0.15.2
torch==2.0.1
tqdm
numpy==1.24.3
deepcs==0.2.6
tensorboard==2.13.0
PySide2
matplotlib
onnx==1.14.0
onnxruntime-gpu==1.15.0
scipy