[#22] parameter changes

eubinecto · Jun 4, 2022 · 18c2f18 · 18c2f18
1 parent de45306
commit 18c2f18
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -8,10 +8,10 @@ kor2eng:
 train:
 	python3 main_train.py \
 	--max_epochs=1000 \
-	--batch_size=128 \
+	--batch_size=256 \
     --save_on_train_epoch_end=1 \
     --every_n_epochs=1 \
-    --log_every_n_steps=10 \
+    --log_every_n_steps=2 \
     --check_val_every_n_epoch=1
 
 train_check:

diff --git a/config.yaml b/config.yaml
@@ -5,7 +5,7 @@ transformer:
   heads: 8
   depth: 5
   max_length: 150
-  lr: 0.1
+  lr: 0.00001
   dropout: 0.1
   tokenizer: "tokenizer:v20"
   # for dataloader