|
|
|
@@ -108,13 +108,13 @@ class CheckpointConfig: |
|
|
|
not keep_checkpoint_max and not keep_checkpoint_per_n_minutes: |
|
|
|
raise ValueError("The input_param can't be all None or 0") |
|
|
|
|
|
|
|
if save_checkpoint_steps: |
|
|
|
if save_checkpoint_steps is not None: |
|
|
|
save_checkpoint_steps = check_int_non_negative(save_checkpoint_steps) |
|
|
|
if save_checkpoint_seconds: |
|
|
|
if save_checkpoint_seconds is not None: |
|
|
|
save_checkpoint_seconds = check_int_non_negative(save_checkpoint_seconds) |
|
|
|
if keep_checkpoint_max: |
|
|
|
if keep_checkpoint_max is not None: |
|
|
|
keep_checkpoint_max = check_int_non_negative(keep_checkpoint_max) |
|
|
|
if keep_checkpoint_per_n_minutes: |
|
|
|
if keep_checkpoint_per_n_minutes is not None: |
|
|
|
keep_checkpoint_per_n_minutes = check_int_non_negative(keep_checkpoint_per_n_minutes) |
|
|
|
|
|
|
|
self._save_checkpoint_steps = save_checkpoint_steps |
|
|
|
|