atropos/environments/community/pytorch_optimizer_coding/FOB/requirements.txt
2025-05-26 14:35:16 +10:00

32 lines
604 B
Text

lightning~=2.2.2
torch~=2.1.1
torchvision~=0.16.1
torchaudio~=2.1.1
torchtext~=0.16.1
tensorboard~=2.15.1
datasets~=2.16.1
pycocotools~=2.0.7
tqdm~=4.66.1
wget~=3.2
deepspeed~=0.12.6
rtdl_revisiting_models~=0.0.2
scikit-learn~=1.5.0
transformers~=4.38.0
tokenizers~=0.15.0
sentencepiece~=0.2.0
sacrebleu~=2.4.1
evaluate~=0.4.1
seaborn~=0.13.1
pandas~=2.2.0
torch_geometric~=2.4.0
spacy~=3.7.2
ogb~=1.3.6
timm~=0.9.12
tensorflow~=2.15.0
tensorflow-datasets~=4.9.4
opencv-python~=4.9.0.80
pytorch-cpr~=0.2.0
mmsegmentation~=1.2.2
mmcv~=2.1.0
lion-pytorch~=0.2.2
git+https://github.com/LiyuanLucasLiu/RAdam