hook_nan_tensor: NaN Loss Monitor

Description Usage Arguments See Also

View source: R/session_run_hooks_builtin_wrappers.R

Description

Monitors loss and stops training if loss is NaN. Can either fail with exception or just stop training.

Usage

1
hook_nan_tensor(loss_tensor, fail_on_nan_loss = TRUE)

Arguments

loss_tensor

The loss tensor.

fail_on_nan_loss

A boolean indicating whether to raise exception when loss is NaN.

See Also

Other session_run_hook wrappers: hook_checkpoint_saver(), hook_global_step_waiter(), hook_history_saver(), hook_logging_tensor(), hook_progress_bar(), hook_step_counter(), hook_stop_at_step(), hook_summary_saver(), session_run_hook()


tfestimators documentation built on Aug. 10, 2021, 1:06 a.m.