public class FederatedPSControlThread extends PSWorker implements Callable<Void>
| Constructor and Description |
|---|
FederatedPSControlThread(int workerID,
String updFunc,
Statement.PSFrequency freq,
Statement.PSRuntimeBalancing runtimeBalancing,
boolean weighting,
int epochs,
long batchSize,
int numBatchesPerGlobalEpoch,
ExecutionContext ec,
ParamServer ps) |
| Modifier and Type | Method and Description |
|---|---|
Void |
call()
Entry point of the functionality
|
String |
getWorkerName() |
void |
setup(double weightingFactor)
Sets up the federated worker and control thread
|
void |
teardown()
cleans up the execution context of the federated worker
|
getFeatures, getLabels, setFeatures, setLabelspublic FederatedPSControlThread(int workerID,
String updFunc,
Statement.PSFrequency freq,
Statement.PSRuntimeBalancing runtimeBalancing,
boolean weighting,
int epochs,
long batchSize,
int numBatchesPerGlobalEpoch,
ExecutionContext ec,
ParamServer ps)
public void setup(double weightingFactor)
weightingFactor - Gradients from this worker will be multiplied by this factor if weighting is enabledpublic void teardown()
public String getWorkerName()
getWorkerName in class PSWorkerCopyright © 2021 The Apache Software Foundation. All rights reserved.