diff --git a/a.yaml b/a.yaml new file mode 100644 index 0000000000000000000000000000000000000000..9f8ce6e7fa889de842853b12c3a723d56ab60899 --- /dev/null +++ b/a.yaml @@ -0,0 +1,83 @@ +common_args: + training_type: "cross_silo" + scenario: "horizontal" + using_mlops: false + config_version: release + name: "exp" + project: "runs/train" + exist_ok: false + random_seed: 0 + + common_args: + training_type: "simulation" + random_seed: 0 +comm_args: + backend: "MPI" + is_mobile: 0 + + +data_args: + dataset: "clintox" + data_cache_dir: ~/fedgraphnn_data/ + part_file: ~/fedgraphnn_data/partition + partition_method: "hetero" + partition_alpha: 0.5 + +model_args: + model: "graphsage" + hidden_size: 32 + node_embedding_dim: 32 + graph_embedding_dim: 64 + readout_hidden_dim: 64 + alpha: 0.2 + num_heads: 2 + dropout: 0.3 + normalize_features: False + normalize_adjacency: False + sparse_adjacency: False + model_file_cache_folder: "./model_file_cache" # will be filled by the server automatically + global_model_file_path: "./model_file_cache/global_model.pt" + +environment_args: + bootstrap: config/bootstrap.sh + +train_args: + federated_optimizer: "FedAvg" + client_id_list: + client_num_in_total: 3 + client_num_per_round: 3 + comm_round: 100 + epochs: 5 + batch_size: 64 + client_optimizer: sgd + learning_rate: 0.03 + weight_decay: 0.001 + metric: "prc-auc" + server_optimizer: sgd + lr: 0.001 + server_lr: 0.001 + wd: 0.001 + ci: 0 + server_momentum: 0.9 + +validation_args: + frequency_of_the_test: 1 + +device_args: + worker_num: 3 + using_gpu: false + gpu_mapping_file: config/gpu_mapping.yaml + gpu_mapping_key: mapping_fedgraphnn_sp + +comm_args: + backend: "MQTT_S3" + mqtt_config_path: config/mqtt_config.yaml + s3_config_path: config/s3_config.yaml + + +tracking_args: + # When running on MLOps platform(open.fedml.ai), the default log path is at ~/.fedml/fedml-client/fedml/logs/ and ~/.fedml/fedml-server/fedml/logs/ + enable_wandb: false + wandb_key: ee0b5f53d949c84cee7decbe7a629e63fb2f8408 + wandb_project: fedml + wandb_name: fedml_torch_moleculenet