From 90b231490df2a04d46090f261ba2d3e31cff5e61 Mon Sep 17 00:00:00 2001 From: Athmane Mansour Bahar <ja_mansourbahar@esi.dz> Date: Thu, 15 Aug 2024 17:51:01 +0000 Subject: [PATCH] Upload New File --- trainer/utils/config.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 trainer/utils/config.py diff --git a/trainer/utils/config.py b/trainer/utils/config.py new file mode 100644 index 0000000..bb9c1ef --- /dev/null +++ b/trainer/utils/config.py @@ -0,0 +1,20 @@ +import argparse + + +def build_args(): + parser = argparse.ArgumentParser(description="MAGIC") + parser.add_argument("--dataset", type=str, default="wget") + parser.add_argument("--device", type=int, default=-1) + parser.add_argument("--lr", type=float, default=0.001, + help="learning rate") + parser.add_argument("--weight_decay", type=float, default=5e-4, + help="weight decay") + parser.add_argument("--negative_slope", type=float, default=0.2, + help="the negative slope of leaky relu for GAT") + parser.add_argument("--mask_rate", type=float, default=0.5) + parser.add_argument("--alpha_l", type=float, default=3, help="`pow`inddex for `sce` loss") + parser.add_argument("--optimizer", type=str, default="adam") + parser.add_argument("--loss_fn", type=str, default='sce') + parser.add_argument("--pooling", type=str, default="mean") + args = parser.parse_args() + return args -- GitLab