From a4e847725c10bb799799fdeda7529ebd595e9a2b Mon Sep 17 00:00:00 2001 From: Linqing Liu Date: Thu, 28 Jun 2018 03:00:42 +0800 Subject: [PATCH] Change default hyperparameters for VDPWI (#130) - Default hyperparameters on SICK now use different batch size, RNN hidden dimension, learning rate, optimizer, and number of epochs. --- vdpwi/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vdpwi/README.md b/vdpwi/README.md index 0ca71fe..5c11a57 100644 --- a/vdpwi/README.md +++ b/vdpwi/README.md @@ -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