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:
|
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']
|
||||||
|
|
|
@ -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']
|
||||||
|
|
Loading…
Reference in New Issue