def test_report_after_finish(session): session.start() session.pause_reporting() session.finish() for _ in range(2): report(loss=1) assert session.get_next() is None
def test_report(): def train(): for i in range(2): report(loss=i) init_session(training_func=train, world_rank=0) session = get_session() session.start() assert session.get_next()["loss"] == 0 assert session.get_next()["loss"] == 1 shutdown_session() with pytest.raises(ValueError): report(loss=2)
def train_2(): for i in range(2): report(loss=i) train_1()
def train(): for i in range(2): report(i) return 1
def train(): for i in range(2): report(loss=i)