diff --git a/config_generate.py b/config_generate.py index 4bafe1e08d78e9aaab6c27038fb8d07274e9ee96..bd719b4186003cb98daae763b49713671287ed5c 100644 --- a/config_generate.py +++ b/config_generate.py @@ -13,7 +13,7 @@ n_segments = "2" random_seed="9527" learning_rate="0.0001" epoches="250" - +drop="0.5" #dir data_dir="./data" @@ -42,6 +42,7 @@ 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","drop",drop) conf.add_section("file_dir") conf.set("file_dir","data",data_dir) @@ -52,4 +53,4 @@ conf.set("device","force",device) conf.set("device","tensor_dtype",default_tensor_type) conf.set("device","confirm_gpu",use_gpu) conf.write(cfg) -cfg.close() \ No newline at end of file +cfg.close()