diff options
Diffstat (limited to 'requirements.txt')
| -rw-r--r-- | requirements.txt | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/requirements.txt b/requirements.txt index deaa0d4..e110c5d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ -torch~=1.7.1 -torchvision~=0.8.0a0+ecf4e9c +torch~=1.7.1+cpu +torch-xla~=1.7 +torchvision~=0.8.2+cpu numpy~=1.19.4 tqdm~=4.56.0 Pillow~=8.1.0 -scikit-learn~=0.23.2
\ No newline at end of file +scikit-learn~=0.24.0
\ No newline at end of file |