torch==1.12.0
torchvision==0.13.0
torchmetrics
onnx
onnxruntime
optuna
wandb
torch-summary
matplotlib