Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "0f3e63eb4dda66ca85357d3cacc0887853922d76"
Commit 1b8c0ee8 authored by Jose Baiocchi's avatar Jose Baiocchi Committed by A. Unique TensorFlower
Browse files

Properly save/restore flags in model tests 

PiperOrigin-RevId: 281192912
parent 021828cc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/utils/testing/integration.py
View file @ 1b8c0ee8
...@@ -25,10 +25,12 @@ import sys ...@@ -25,10 +25,12 @@ import sys
import tempfile import tempfile
from absl import flags from absl import flags
from absl.testing import flagsaver
from official.utils.flags import core as flags_core from official.utils.flags import core as flags_core
@flagsaver.flagsaver
def run_synthetic(main, tmp_root, extra_flags=None, synth=True, train_epochs=1, def run_synthetic(main, tmp_root, extra_flags=None, synth=True, train_epochs=1,
epochs_between_evals=1): epochs_between_evals=1):
"""Performs a minimal run of a model. """Performs a minimal run of a model.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment