SERVER_URL = "http://127.0.0.1:9763"
CLIENT_IP = "127.0.0.1"
CLIENT_PORT = 8081
CLIENT_ID = 0
if __name__ == "__main__":
mnist_data = datasets.MNIST("./mnist_data",
download=True,
train=True,
transform=transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.13066062, ),
(0.30810776, ))
]))
client = FLClient()
pfl_models = client.get_remote_pfl_models(SERVER_URL)
for pfl_model in pfl_models:
optimizer = torch.optim.SGD(pfl_model.get_model().parameters(),
lr=0.01,
momentum=0.5)
train_strategy = TrainStrategy(optimizer=optimizer,
batch_size=32,
loss_function=LossStrategy.NLL_LOSS)
pfl_model.set_train_strategy(train_strategy)
TrainerController(work_mode=WorkModeStrategy.WORKMODE_CLUSTER,
models=pfl_models,
data=mnist_data,
client_id=CLIENT_ID,
from gfl.core.client import FLClient
from gfl.core.strategy import WorkModeStrategy, TrainStrategy, LossStrategy
from gfl.core.trainer_controller import TrainerController
CLIENT_ID = 1
if __name__ == "__main__":
# CLIENT_ID = int(sys.argv[1])
dataset_path = os.path.join(os.path.abspath("../"), "cifa10_demo",
"cifa10", "train_dataset_dir",
"train_dataset_{}".format(CLIENT_ID))
dataset = torch.load(dataset_path)
client = FLClient()
pfl_models = client.get_remote_pfl_models()
for pfl_model in pfl_models:
optimizer = torch.optim.SGD(pfl_model.get_model().parameters(),
lr=0.001,
momentum=0.9)
train_strategy = TrainStrategy(optimizer=optimizer,
batch_size=32,
loss_function=LossStrategy.NLL_LOSS)
pfl_model.set_train_strategy(train_strategy)
TrainerController(work_mode=WorkModeStrategy.WORKMODE_STANDALONE,
models=pfl_models,
data=dataset,
client_id=CLIENT_ID,
import os
import torch
from torchvision import datasets, transforms
from gfl.core.client import FLClient
from gfl.core.strategy import WorkModeStrategy, TrainStrategy, LossStrategy
from gfl.core.trainer_controller import TrainerController
CLIENT_ID = 0
if __name__ == "__main__":
# CLIENT_ID = int(sys.argv[1])
dataset_path = os.path.join(os.path.abspath("../"), "cifa10_demo", "cifa10", "train_dataset_dir",
"train_dataset_{}".format(CLIENT_ID))
dataset = torch.load(dataset_path)
client = FLClient()
gfl_models = client.get_remote_gfl_models()
for gfl_model in gfl_models:
optimizer = torch.optim.SGD(gfl_model.get_model().parameters(), lr=0.01, momentum=0.5)
train_strategy = TrainStrategy(optimizer=optimizer, batch_size=32, loss_function=LossStrategy.NLL_LOSS)
gfl_model.set_train_strategy(train_strategy)
TrainerController(work_mode=WorkModeStrategy.WORKMODE_STANDALONE, models=gfl_models, data=dataset, client_id=CLIENT_ID,
curve=True, local_epoch=5, concurrent_num=3).start()