Skip to content

GitLab

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
import tempfile
from absl import flags
from absl.testing import flagsaver
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,
epochs_between_evals=1):
"""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