Merge pull request #18 from KernelA/Fix-description

Update README.md
KernelA · Aug 8, 2023 · 6935383 · 6935383
2 parents 7ebcc19 + 3eba321
commit 6935383
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -33,14 +33,14 @@ You can download a trained model from the [model registry](https://docs.wandb.ai
 
 ### CPU (training will be slow)
 ```
-pip install -r ./requirements.torch.cpu.txt
+pip install -r ./requirements.torch.gpu.txt
 pip install -r ./requirements.cpu.txt -r ./requirements.base.txt
 ```
 
 ### GPU:
 ```
-pip install -r ./requirements.torch.cpu.txt
-pip install -r ./requirements.cpu.txt -r ./requirements.base.txt
+pip install -r ./requirements.torch.gpu.txt
+pip install -r ./requirements.gpu.txt -r ./requirements.base.txt
 ```
 
 Only for `presentation.ipynb`