torch==1.12
pytorch_lightning==1.7
scikit-learn
protobuf==3.19.1
pandas==1.3
