Python has_training_endedの例

コード例 #1

def test_gradtape_training_end(out_dir):
    """
    Verify enf of training file
    """
    helper_keras_gradtape(out_dir,
                          include_collections=[CollectionKeys.OUTPUTS])
    assert has_training_ended(out_dir) is True

コード例 #2

def test_training_end(out_dir, tf_eager_mode):
    helper_keras_fit(
        out_dir,
        include_collections=[CollectionKeys.OUTPUTS],
        steps=["train"],
        run_eagerly=tf_eager_mode,
    )
    assert has_training_ended(out_dir) is True

コード例 #3

ファイル: test_keras.py プロジェクト: brightsparc/sagemaker-debugger

def test_training_end(out_dir):
    train_model(out_dir,
                include_collections=[CollectionKeys.OUTPUTS],
                steps=["train"])
    assert has_training_ended(out_dir) is True