mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
57db4640ca
Pulls out Tune's MLflow logging logic to a shared MLflow util. Adds an MLflow logger callback to Ray Train Closes #20642
11 lines
247 B
Text
11 lines
247 B
Text
-r requirements_dl.txt
|
|
|
|
mlflow==1.21.0
|
|
tensorboardX==2.4.1
|
|
|
|
# Dependencies for Hugging Face examples:
|
|
# `python/ray/train/examples/transformers/transformers_example.py`
|
|
transformers==4.10.0
|
|
accelerate==0.5.1
|
|
datasets==1.14.0
|
|
sentencepiece==0.1.96
|