wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change default hyperparameters for VDPWI (#130)

Browse Source 
- Default hyperparameters on SICK now use different batch size, RNN hidden dimension, learning rate, optimizer, and number of epochs.
This commit is contained in:
Linqing Liu
2018-06-27 15:00:42 -04:00
committed by Ralph Tang
parent e4149ff400
commit a4e847725c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vdpwi/README.md
+1 -1
View File
@@ -12,7 +12,7 @@ Please ensure you have followed instructions in the main [README](../README.md)
To run VDPWI on the SICK dataset, use the following command. If you have any problems running it check the Troubleshooting section below.
```
python -m vdpwi vdpwi.sick.model.castor --dataset sick --epochs 19 --epsilon 1e-7
python -m vdpwi vdpwi.sick.model.castor --dataset sick --lr 5e-4 --optimizer rmsprop --epochs 15 --batch-size 8 --rnn-hidden-dim 256 --epsilon 1e-7
```
## MSRVID Dataset