server=Server,
dataset='cifar10',
avg_logic='standalone',
num_clients=20,
comm_rounds=50,
frac=1,
prune_step=0.2,
prune_percent=0.6,
acc_thresh=0.5,
client_epoch=10,
batch_size=32,
lr=0.001,
rate_unbalance=1.0,
n_samples=20,
n_class=2,
eita_hat=0.5,
alpha=0.5,
prune_threshold=0.1,
report_verbosity=False,
train_verbosity=False,
test_verbosity=False,
globalPrune=True,
prune_verbosity=True,
global_prune_freq=25,
save_freq=10)
}
if __name__ == "__main__":
overrides = {'log_folder': './report_output', 'running_on_cloud': False}
run_experiments(CIFAR10_experiments, overrides)
add_experiment(experiments, traffic, '', '', max_cycles, num_nodes,
'OddEven', 'BufferLevel', data_packet_injection_rate,
-1, -1)
for aco_selection_alpha in aco_selection_alpha_range:
for reinforcement_factor in reinforcement_factor_range:
add_experiment(experiments, traffic, '', '', max_cycles,
num_nodes, 'OddEven', 'ACO',
data_packet_injection_rate, aco_selection_alpha,
reinforcement_factor)
for bench, trace_file_name in bench_and_trace_file_name_range:
for data_packet_injection_rate in trace_driven_data_packet_injection_rate_range:
add_experiment(experiments, 'Trace', bench,
trace_file_name + '.combined', max_cycles, num_nodes,
'XY', 'Random', data_packet_injection_rate, -1, -1)
add_experiment(experiments, 'Trace', bench,
trace_file_name + '.combined', max_cycles, num_nodes,
'OddEven', 'BufferLevel', data_packet_injection_rate,
-1, -1)
for aco_selection_alpha in aco_selection_alpha_range:
for reinforcement_factor in reinforcement_factor_range:
add_experiment(experiments, 'Trace', bench,
trace_file_name + '.combined', max_cycles,
num_nodes, 'OddEven', 'ACO',
data_packet_injection_rate, aco_selection_alpha,
reinforcement_factor)
run_experiments(experiments, run_experiment)