def start(self, iter_id):
assert self._current_iter_id is None, "Last iter not finished"
self._current_iter_id = iter_id
msg = tws_pb.TrainerWorkerMessage(start=tws_pb.StartMessage(
iter_id=iter_id))
self._transmit(msg)
logging.debug("Starting iter %d", iter_id)
def fake_start_message(seq_num, iter_id):
return tws_pb.TrainerWorkerMessage(
seq_num=seq_num, start=tws_pb.StartMessage(iter_id=iter_id))
