From 36a11d271073e7722e0a3575ec9c5a830c8ea430 Mon Sep 17 00:00:00 2001 From: Athmane Mansour Bahar <ja_mansourbahar@esi.dz> Date: Thu, 15 Aug 2024 17:47:21 +0000 Subject: [PATCH] Upload New File --- utils/config.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 utils/config.py diff --git a/utils/config.py b/utils/config.py new file mode 100644 index 0000000..ee14920 --- /dev/null +++ b/utils/config.py @@ -0,0 +1,13 @@ +def build_args(): + args = {} + args["dataset"] ="wget" + args["device"]=-1 + args["lr"]=0.001 + args["weight_decay"]=5e-4 + args["negative_slope"]=0.2 + args["mask_rate"]=0.5 + args["alpha_l"]=3 + args["optimizer"]="adam" + args["loss_fn"]='sce' + args["pooling"] = "mean" + return args -- GitLab