unilab.training.run.apply_env_nan_guard

unilab.training.run.apply_env_nan_guard(env, training_cfg)[source]

Attach a NanGuard to env when training.nan_guard is enabled.

Parameters:

  • env (Any)

  • training_cfg (Any)

Return type:

None