def _setup_initial_blobs(self):
self._update_rate_blob = self.model_id + "_update_rate"
workspace.FeedBlob(self._update_rate_blob,
np.array([1], dtype=np.float32))
self._retain_rate_blob = self.model_id + "_retain_rate"
workspace.FeedBlob(self._retain_rate_blob,
np.array([0], dtype=np.float32))
CNN._setup_initial_blobs(self)
def __init__(
self,
name: str,
cnn_parameters: CNNParameters,
target_update_rate: float,
source_trainer: ConvMLTrainer,
) -> None:
self._target_update_rate = target_update_rate
self.enabled_slow_updates = False
CNN.__init__(self, name, cnn_parameters)
self._setup_update_net(source_trainer)
workspace.RunNetOnce(self._update_model.param_init_net)
workspace.CreateNet(self._update_model.net)
def __init__(
self,
name: str,
cnn_parameters: CNNParameters,
) -> None:
CNN.__init__(self, name, cnn_parameters)