MCPcopy Create free account
hub / github.com/apache/singa / set_params

Method set_params

python/singa/layer.py:125–138  ·  view source on GitHub ↗

Set parameters for this layer and all sublayers Args: parameters(dict): A dictionary contains parameter names and corresponding values. The value shoud be either a PyTensor or numpy ndarray

(self, parameters)

Source from the content-addressed store, hash-verified

123 return params
124
125 def set_params(self, parameters):
126 """ Set parameters for this layer and all sublayers
127
128 Args:
129 parameters(dict): A dictionary contains parameter names
130 and corresponding values. The value shoud be either a
131 PyTensor or numpy ndarray
132 """
133 names = parameters.keys()
134 sublayers = self._layers
135 for name, sublayer in sublayers.items():
136 if sublayer._initialized:
137 if self._has_layer_param(sublayer, names):
138 sublayer.set_params(parameters)
139
140 def get_states(self):
141 """ Get states of this layer and all sublayers

Callers 1

set_statesMethod · 0.95

Calls 2

_has_layer_paramMethod · 0.95
set_paramsMethod · 0.45

Tested by

no test coverage detected