Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
yikang-li committed Nov 19, 2019
1 parent 13e74a9 commit 13c9b38
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Expand Up @@ -78,7 +78,7 @@ ln -s /path/to/downloaded_folder top_150_50

For full Dataset Training:
```
CUDA_VISIBLE_DEVICES=0 python train_rpn_region.py --max_epoch=10 --step_size=2 --dataset_option=normal --model_name=RPN_full_region
CUDA_VISIBLE_DEVICES=0 python train_rpn.py --max_epoch=10 --step_size=2 --dataset_option=normal --model_name=RPN_full_region
```

```--step_size``` is set to indicate the number of epochs to decay the learning rate, ```dataset_option``` is to indicate the ```\[ small | fat | normal \]``` subset.
Expand Down

0 comments on commit 13c9b38

Please sign in to comment.