Skip to content
Snippets Groups Projects
config.ini 510 B
Newer Older
Rishi Sharma's avatar
Rishi Sharma committed
[GRAPH]
package = decentralizepy.graphs.SmallWorld
graph_class = SmallWorld
Rishi Sharma's avatar
Rishi Sharma committed

[DATASET]
dataset_package = decentralizepy.datasets.Femnist
dataset_class = Femnist
model_class = LogisticRegression
n_procs = 1.0
train_dir = 
test_dir = 
; python list of fractions below
sizes = [0.4, 0.2, 0.3, 0.1]
Rishi Sharma's avatar
Rishi Sharma committed

[OPTIMIZER_PARAMS]
optimizer_package = torch.optim
optimizer_class = SGD
lr = 0.1

[TRAIN_PARAMS]
training_package = decentralizepy.training.Training
training_class = Training
epochs_per_round = 25
batch_size = 64