diff --git a/scheduler/requirements.txt b/scheduler/requirements.txt index 80bb58c3..6551ecdb 100644 --- a/scheduler/requirements.txt +++ b/scheduler/requirements.txt @@ -18,6 +18,6 @@ preconditions psutil seaborn setproctitle -sklearn +scikit-learn torch==1.4.0 torchvision==0.5.0