Remove unneeded Saver instances
This commit is contained in:
parent
0e2f34f8cf
commit
bd8b96c19d
|
@ -811,9 +811,6 @@ def do_single_file_inference(input_file_path):
|
|||
with tfv1.Session(config=Config.session_config) as session:
|
||||
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
|
||||
if FLAGS.load == 'auto':
|
||||
method_order = ['best', 'last']
|
||||
|
|
|
@ -81,9 +81,6 @@ def evaluate(test_csvs, create_model):
|
|||
except NotImplementedError:
|
||||
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:
|
||||
if FLAGS.load == 'auto':
|
||||
method_order = ['best', 'last']
|
||||
|
|
Loading…
Reference in New Issue