Update README.md

1fd8e1a1 · Yizhou Wang · 1d3dead7 · 1fd8e1a1
Commit 1fd8e1a1 authored Dec 30, 2020 by Yizhou Wang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 6 deletions

README.md README.md +1 -6

No files found.
--- a/README.md
+++ b/README.md
@@ -10,7 +10,7 @@ conda activate rodnet

 Install pytorch.
 ```
-conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
+conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
 ```

 Setup RODNet package.
@@ -36,11 +36,6 @@ python tools/train.py --config configs/<CONFIG_FILE> \
        --log_dir checkpoints/
 ```

-### Pretrained Models
-
-Please download the pretrained models from the list below:
- [rodnet_cdc_win16_cruw_mini_epoch50.pkl](https://drive.google.com/file/d/19j-jy5wVOwYjUS5SPvoF-MjA06PepYnA/view?usp=sharing)
-
 ## Inference

 ```