Skip to content

Commit

Permalink
Add cuda example
Browse files Browse the repository at this point in the history
  • Loading branch information
bacox committed Mar 17, 2022
1 parent 2a22902 commit 672f926
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 0 deletions.
25 changes: 25 additions & 0 deletions experiments/example_cuda/descr.yaml
Original file line number Diff line number Diff line change
@@ -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
5 changes: 5 additions & 0 deletions experiments/example_cuda/fedavg.cfg.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
# Individual configuration
offload_stategy: vanilla
deadline: 500
single_machine: true
real_time: false

0 comments on commit 672f926

Please sign in to comment.