463465810cz commited on
Commit ·
6e0dcf2
1
Parent(s): 4163a01
DAT
Browse files- README.md +1 -1
- requirements.txt +1 -1
README.md
CHANGED
|
@@ -5,7 +5,7 @@ This repository is for DAT introduced in the paper.
|
|
| 5 |
## Dependencies
|
| 6 |
|
| 7 |
- Python 3.8
|
| 8 |
-
-
|
| 9 |
- NVIDIA GPU + [CUDA](https://developer.nvidia.com/cuda-downloads)
|
| 10 |
|
| 11 |
```bash
|
|
|
|
| 5 |
## Dependencies
|
| 6 |
|
| 7 |
- Python 3.8
|
| 8 |
+
- PyTorch 1.8.0
|
| 9 |
- NVIDIA GPU + [CUDA](https://developer.nvidia.com/cuda-downloads)
|
| 10 |
|
| 11 |
```bash
|
requirements.txt
CHANGED
|
@@ -9,7 +9,7 @@ requests
|
|
| 9 |
scikit-image
|
| 10 |
scipy
|
| 11 |
tb-nightly
|
| 12 |
-
torch
|
| 13 |
torchvision
|
| 14 |
tqdm
|
| 15 |
yapf
|
|
|
|
| 9 |
scikit-image
|
| 10 |
scipy
|
| 11 |
tb-nightly
|
| 12 |
+
torch==1.8.0
|
| 13 |
torchvision
|
| 14 |
tqdm
|
| 15 |
yapf
|