Update environment.yml

add torchmetrics
This commit is contained in:
curious-energy 2024-01-24 16:54:45 +08:00 committed by GitHub
parent fea81bb550
commit a050ffceb0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -9,7 +9,7 @@ dependencies:
- python=3.7.13 - python=3.7.13
- pip=22.3.1 - pip=22.3.1
- pytorch=1.12.1 - pytorch=1.12.1
- torchaudio=0.12.1 - torchmetrics
- torchvision=0.13.1 - torchvision=0.13.1
- tqdm - tqdm
- pip: - pip: