poster2plot / train /requirements.txt
dsr's picture
Add dataset creation and model training code
05293b5
raw
history blame
170 Bytes
--find-links https://download.pytorch.org/whl/torch_stable.html
pandas==1.3.4
scikit-learn==1.0.1
python-box==5.4.1
transformers==4.12.5
torch==1.10.0+cu113
Pillow==8.4.0