Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
@Sierkinhane
Sierkinhane committed Jul 14, 2023
1 parent 8b1dcbc commit 78448d4
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions WSOL/PSOL/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,6 @@ bash run_ilsvrc.sh
```

**Please note that** we specify ResNet50 as default backbone to prevent the problem of OOM. If you have enough GPUs, please specify more gpu numbers in 'run_xx.sh'. In our experiments, we use **two tesla A100** to train a regressor with a backbone of DenseNet161.

**Pretrained weights are available at [here](https://drive.google.com/drive/folders/1Uoyh3XLdz5VqWvWNjxoqMfRryu1nkQTL?usp=sharing)**

0 comments on commit 78448d4

Please sign in to comment.