Description Usage Arguments See Also
View source: R/session_run_hooks_builtin_wrappers.R
Monitors loss and stops training if loss is NaN. Can either fail with exception or just stop training.
1 | hook_nan_tensor(loss_tensor, fail_on_nan_loss = TRUE)
|
loss_tensor |
The loss tensor. |
fail_on_nan_loss |
A boolean indicating whether to raise exception when loss is NaN. |
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()
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.