public class TimeDelayedParameterUpdater extends BaseParameterUpdater
ndArrayHolder, updateStorage| Constructor and Description |
|---|
TimeDelayedParameterUpdater() |
| Modifier and Type | Method and Description |
|---|---|
void |
partialUpdate(org.nd4j.linalg.api.ndarray.INDArray arr,
org.nd4j.linalg.api.ndarray.INDArray result,
long idx,
int... dimensions)
Updates result
based on arr along a particular
INDArray#tensorAlongDimension(int, int...) |
int |
requiredUpdatesForPass()
Returns the number of required
updates for a new pass
|
void |
reset()
Reset internal counters
such as number of updates accumulated.
|
boolean |
shouldReplicate()
Returns true if
the updater has accumulated enough ndarrays to
replicate to the workers
|
Map<String,Number> |
status()
Returns the current status of this parameter server
updater
|
String |
toJson()
Serialize this updater as json
|
void |
update(org.nd4j.linalg.api.ndarray.INDArray arr,
org.nd4j.linalg.api.ndarray.INDArray result)
Updates result
based on arr
|
void |
update(NDArrayMessage message)
Do an update based on the ndarray message.
|
isAsync, isReady, ndArrayHolder, numUpdatespublic int requiredUpdatesForPass()
public Map<String,Number> status()
public String toJson()
public void reset()
reset in interface ParameterServerUpdaterreset in class BaseParameterUpdaterpublic boolean shouldReplicate()
public void update(NDArrayMessage message)
message - public void partialUpdate(org.nd4j.linalg.api.ndarray.INDArray arr, org.nd4j.linalg.api.ndarray.INDArray result, long idx, int... dimensions)
INDArray#tensorAlongDimension(int, int...)arr - the array to updateresult - the result ndarray to updateidx - the index to updatedimensions - the dimensions to updatepublic void update(org.nd4j.linalg.api.ndarray.INDArray arr, org.nd4j.linalg.api.ndarray.INDArray result)
arr - the array to updateresult - the result ndarray to updateCopyright © 2021. All rights reserved.