diff --git a/requirements.txt b/requirements.txt index 09f4807..029b2ec 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,5 +2,5 @@ numpy==1.18.1 torch==1.4.0 torchvision==0.5.0 matplotlib==3.1.3 -scikit_learn==0.22.1 +scikit_learn==1.5.0 tensorboardX==2.0