diff --git a/utils/config.py b/utils/config.py new file mode 100644 index 0000000000000000000000000000000000000000..ee14920f4b2e36dfcd0d7fddca100572837ade20 --- /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