udpate gatne config file

Author: Liwb5

apps/Graph4Rec/user_configs/gatne.yaml

@@ -0,0 +1,59 @@
+# configuration for GATNE
+
+task_name: gatne.0712
+
+# ------------------------Data Configuration--------------------------------------------#
+etype2files: "u2buy2t:./toy_data/u2buy2i.txt,u2click2i:./toy_data/u2click2i.txt"
+ntype2files: "u:./toy_data/node_types.txt,i:./toy_data/node_types.txt"
+symmetry: True
+shard_num: 1000
+# [ntype, name, feat_type, length]
+nfeat_info: null
+slots: []
+
+meta_path: "u2buy2i-i2buy2u;u2click2i-i2click2u;i2buy2u-u2buy2i;i2click2u-u2click2i"
+
+
+walk_len: 4
+win_size: 3
+neg_num: 10
+walk_times: 10
+
+
+# -----------------Model HyperParams Configuration--------------------------------------#
+dataset_type: SageDataset
+collatefn: SageCollateFn
+model_type: SageModel
+warm_start_from: null
+num_nodes: 5500
+embed_size: 64
+hidden_size: 64
+res_alpha: 0.9
+
+sage_mode: True
+sage_layer_type: LightGCNConv
+sage_act: null
+interact_mode: gatne
+sample_num_list: [10]
+infer_sample_num_list: [100]
+
+# ----------------------Training Configuration------------------------------------------#
+epochs: 10
+num_workers: 1
+lr: 0.001
+lazy_mode: True
+batch_node_size: 20
+batch_pair_size: 100
+pair_stream_shuffle_size: 10000
+log_dir: ../../logs_custom
+output_dir: ../../outputs_custom
+save_dir: ../../ckpt_custom
+files2saved: ["*.yaml", "*.py", "*.sh", "./models", "./datasets", "./utils"]
+log_steps: 100
+
+# -------------Distributed CPU Training Configuration-----------------------------------#
+
+# if you want to save model per epoch, then save_steps will be set by below equation
+# save_steps = num_nodes * walk_len * win_size * walk_times / batch_pair_size / worker_num
+# but the equation is not very precise since the neighbors of each node is not the same.
+save_steps: 100000