From abcb87ad907cf3d901f8238005178da741199d77 Mon Sep 17 00:00:00 2001 From: Stevezhangz Date: Fri, 23 Apr 2021 02:53:58 +0800 Subject: [PATCH] Update config_generate.py --- config_generate.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/config_generate.py b/config_generate.py index bd719b4..3f5f562 100644 --- a/config_generate.py +++ b/config_generate.py @@ -1,18 +1,19 @@ import configparser # param -maxlen = "150" -batch_size = "6" +maxlen = "120" +batch_size = "12" max_pred = "5" n_layers = "6" n_heads = "12" -d_model = "768" -d_ff = "768*4" -d_k = d_v = "64" +d_model = "1000" +d_ff = "1000*4" +d_k = "100" +d_v = "100" n_segments = "2" -random_seed="9527" -learning_rate="0.0001" -epoches="250" +seed = "9527" +lr = "0.0001" +epc = "250" drop="0.5" #dir @@ -39,9 +40,9 @@ conf.set("hyper_parameters","d_ff",d_ff) conf.set("hyper_parameters","d_k",d_k) conf.set("hyper_parameters","d_v",d_v ) conf.set("hyper_parameters","n_segments",n_segments) -conf.set("hyper_parameters","seed",random_seed) -conf.set("hyper_parameters","lr",learning_rate) -conf.set("hyper_parameters","epc",epoches) +conf.set("hyper_parameters","seed",seed) +conf.set("hyper_parameters","lr",lr) +conf.set("hyper_parameters","epc",epc) conf.set("hyper_parameters","drop",drop) conf.add_section("file_dir") -- GitLab