diff --git a/experiments/example_cuda/descr.yaml b/experiments/example_cuda/descr.yaml new file mode 100644 index 00000000..b42db5f4 --- /dev/null +++ b/experiments/example_cuda/descr.yaml @@ -0,0 +1,25 @@ +--- +# Experiment configuration +total_epochs: 3 +rounds: 5 +epochs_per_cycle: 1 +wait_for_clients: true +net: MNISTCNN +dataset: mnist +# Use cuda is available; setting to false will force CPU +cuda: true +profiling_time: 100 +warmup_round: false +output_location: 'output/example_cuda' +tensor_board_active: true +clients_per_round: 2 +node_groups: + slow: [1, 1] + medium: [2, 2] + fast: [3, 3] +sampler: "uniform" # "limit labels" || "q sampler" || "dirichlet" || "uniform" (default) +sampler_args: + - 0.07 # label limit || q probability || alpha || unused + - 42 # random seed || random seed || random seed || unused +num_clients: 10 +replications: 5 diff --git a/experiments/example_cuda/fedavg.cfg.yaml b/experiments/example_cuda/fedavg.cfg.yaml new file mode 100644 index 00000000..25a64bda --- /dev/null +++ b/experiments/example_cuda/fedavg.cfg.yaml @@ -0,0 +1,5 @@ +# Individual configuration +offload_stategy: vanilla +deadline: 500 +single_machine: true +real_time: false \ No newline at end of file