def load_weights(self, prefix):
weights = self._server.get(f"{self._prefix}_{prefix}_weights")
if weights is None:
return None
weights = unpack(weights)
self._epoch = weights.get("epoch")
return weights["weights"]
def load_weights(self, prefix):
weights_obj = self._weights_collection.find_one({"prefix": prefix})
weights = weights_obj.get("weights")
if weights is None:
return None
self._epoch = weights_obj["epoch"]
weights = unpack(weights)
return weights
def get_trajectory(self, index=None):
index = index if index is not None else self._index
trajectory = self._server.lindex("trajectories", index)
if trajectory is not None:
self._index = index + 1
trajectory = unpack(trajectory)
trajectory, trajectory_epoch = \
trajectory["trajectory"], trajectory["epoch"]
if self._sync_epoch and self._epoch != trajectory_epoch:
trajectory = None
return trajectory
def get_trajectory(self, index=None):
assert index is None
trajectory_obj = self._trajectory_collection.find_one(
{"date": {
"$gt": self._last_datetime
}})
if trajectory_obj is not None:
self._last_datetime = trajectory_obj["date"]
trajectory, trajectory_epoch = \
unpack(trajectory_obj["trajectory"]), trajectory_obj["epoch"]
if self._sync_epoch and self._epoch != trajectory_epoch:
trajectory = None
else:
trajectory = None
return trajectory