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
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
def test_training_end(out_dir): train_model(out_dir, include_collections=[CollectionKeys.OUTPUTS], steps=["train"]) assert has_training_ended(out_dir) is True