Skip to content
Snippets Groups Projects
config_femnist_partialmodel.ini 809 B
Newer Older
Rishi Sharma's avatar
Rishi Sharma committed
[DATASET]
dataset_package = decentralizepy.datasets.Femnist
dataset_class = Femnist
Jeffrey Wigger's avatar
Jeffrey Wigger committed
random_seed = 97
Rishi Sharma's avatar
Rishi Sharma committed
model_class = CNN
Jeffrey Wigger's avatar
Jeffrey Wigger committed
train_dir = /mnt/nfs/shared/leaf/data/femnist/per_user_data/train
test_dir = /mnt/nfs/shared/leaf/data/femnist/data/test
; python list of fractions below
Rishi Sharma's avatar
Rishi Sharma committed
sizes = 
Rishi Sharma's avatar
Rishi Sharma committed

[OPTIMIZER_PARAMS]
optimizer_package = torch.optim
Rishi Sharma's avatar
Rishi Sharma committed
optimizer_class = Adam
lr = 0.001
training_package = decentralizepy.training.Training
training_class = Training
Jeffrey Wigger's avatar
Jeffrey Wigger committed
rounds = 47
full_epochs = False
Jeffrey Wigger's avatar
Jeffrey Wigger committed
batch_size = 16
Rishi Sharma's avatar
Rishi Sharma committed
shuffle = True
loss_package = torch.nn
loss_class = CrossEntropyLoss
Rishi Sharma's avatar
Rishi Sharma committed

[COMMUNICATION]
Rishi Sharma's avatar
Rishi Sharma committed
comm_package = decentralizepy.communication.TCP
comm_class = TCP
Rishi Sharma's avatar
Rishi Sharma committed
addresses_filepath = ip_addr_6Machines.json
Rishi Sharma's avatar
Rishi Sharma committed

[SHARING]
Rishi Sharma's avatar
Rishi Sharma committed
sharing_package = decentralizepy.sharing.PartialModel
sharing_class = PartialModel
Jeffrey Wigger's avatar
Jeffrey Wigger committed
alpha=0.1