Remove unneeded Saver instances

This commit is contained in:
Reuben Morais 2020-02-17 12:46:32 +01:00
parent 0e2f34f8cf
commit bd8b96c19d
2 changed files with 0 additions and 6 deletions

View File

@ -811,9 +811,6 @@ def do_single_file_inference(input_file_path):
with tfv1.Session(config=Config.session_config) as session: with tfv1.Session(config=Config.session_config) as session:
inputs, outputs, _ = create_inference_graph(batch_size=1, n_steps=-1) inputs, outputs, _ = create_inference_graph(batch_size=1, n_steps=-1)
# Create a saver using variables from the above newly created graph
saver = tfv1.train.Saver()
# Restore variables from training checkpoint # Restore variables from training checkpoint
if FLAGS.load == 'auto': if FLAGS.load == 'auto':
method_order = ['best', 'last'] method_order = ['best', 'last']

View File

@ -81,9 +81,6 @@ def evaluate(test_csvs, create_model):
except NotImplementedError: except NotImplementedError:
num_processes = 1 num_processes = 1
# Create a saver using variables from the above newly created graph
saver = tfv1.train.Saver()
with tfv1.Session(config=Config.session_config) as session: with tfv1.Session(config=Config.session_config) as session:
if FLAGS.load == 'auto': if FLAGS.load == 'auto':
method_order = ['best', 'last'] method_order = ['best', 'last']