From 9cd3698b2921bb67ac489128d73736d94889d24a Mon Sep 17 00:00:00 2001 From: Stevezhangz Date: Fri, 23 Apr 2021 02:35:37 +0800 Subject: [PATCH] Update Config_load.py --- Config_load.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/Config_load.py b/Config_load.py index 3887de3..e34bdca 100644 --- a/Config_load.py +++ b/Config_load.py @@ -22,4 +22,5 @@ data_dir=conf.get("file_dir","data") weight_dir=conf.get("file_dir","data") device=conf.get("device","force") default_tensor_type=conf.get("device","tensor_dtype") -use_gpu=int(conf.get("device","confirm_gpu")) \ No newline at end of file +use_gpu=int(conf.get("device","confirm_gpu")) +drop=float(conf.get("hyper_parameters","drop")) -- GitLab