Update requirements.txt

nateagr / clip-on-yarn

Distributed training of CLIP on Yarn

7 stars 2 forks source link

Open jcuquemelle opened 1 year ago

jcuquemelle commented 1 year ago

Limit version of tf-yarn as there are 2 breaking changes in the last version:

removal of PYTORCH_DDP_RANK usage in favor of explicitly checking rank before checkpointing
passing number of tasks per container in the TaskSpecs instead of the Experiment