fedscale.core package

Subpackages

Submodules

fedscale.core.client_manager module

class fedscale.core.client_manager.clientManager(mode, args, sample_seed=233)[source]

Bases: object

clientOnHost(clientIds, hostId)[source]
clientSampler(clientId)[source]
getAllClients()[source]
getAllClientsLength()[source]
getAllMetrics()[source]
getClient(clientId)[source]
getClientLenOnHost(hostId)[source]
getClientReward(clientId)[source]
getClientSize(clientId)[source]
getClientsInfo()[source]
getCompletionTime(clientId, batch_size, upload_step, upload_size, download_size)[source]
getCurrentClientIds(hostId)[source]
getDataInfo()[source]
getFeasibleClients(cur_time)[source]
getSampleRatio(clientId, hostId, even=False)[source]
getScore(hostId, clientId)[source]
getUniqueId(hostId, clientId)[source]
get_median_reward()[source]
isClientActive(clientId, cur_time)[source]
nextClientIdToRun(hostId)[source]
registerClient(hostId, clientId, size, speed, duration=1)[source]
registerClientScore(clientId, reward)[source]
registerDuration(clientId, batch_size, upload_step, upload_size, download_size)[source]
registerScore(clientId, reward, auxi=1.0, time_stamp=0, duration=1.0, success=True)[source]
registerSpeed(hostId, clientId, speed)[source]
resampleClients(numOfClients, cur_time=0)[source]

fedscale.core.commons module

fedscale.core.config_parser module

fedscale.core.fllibs module

fedscale.core.fllibs.init_dataset()[source]
fedscale.core.fllibs.init_model()[source]

fedscale.core.resource_manager module

class fedscale.core.resource_manager.ResourceManager(experiment_mode)[source]

Bases: object

Schedule training tasks across GPUs/CPUs

get_next_task(client_id=None)[source]
has_next_task(client_id=None)[source]
register_tasks(clientsToRun)[source]
remove_client_task(client_id)[source]

Module contents