-
Notifications
You must be signed in to change notification settings - Fork 70
Open
Description
i try to run batch prediction with command:
"python deepdenoiser/predict.py --format=mseed --model_dir=model/190614-104802 --data_dir=test_data/ --data_list=test_data/testing.csv --sampling_rate=100 --plot_figure --save_signal --save_noise"
but getting an error:
Traceback (most recent call last):
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 136, in <module>
main(args)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 129, in main
pred_fn(args, data_reader, log_dir=args.output_dir)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 81, in pred_fn
X_batch, fname_batch, t0_batch = sess.run(data_batch)
File "D:\miniconda3\lib\site-packages\tensorflow\python\client\session.py", line 968, in run
result = self._run(None, fetches, feed_dict, options_ptr,
File "D:\miniconda3\lib\site-packages\tensorflow\python\client\session.py", line 1191, in _run
results = self._do_run(handle, final_targets, final_fetches,
File "D:\miniconda3\lib\site-packages\tensorflow\python\client\session.py", line 1371, in _do_run
return self._do_call(_run_fn, feeds, fetches, targets, options,
File "D:\miniconda3\lib\site-packages\tensorflow\python\client\session.py", line 1397, in _do_call
raise type(e)(node_def, op, message) # pylint: disable=no-value-for-parameter
tensorflow.python.framework.errors_impl.InvalidArgumentError: Graph execution error:
Detected at node 'Input_Batch/IteratorGetNext' defined at (most recent call last):
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 136, in <module>
main(args)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 129, in main
pred_fn(args, data_reader, log_dir=args.output_dir)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 56, in pred_fn
data_batch = data_reader.dataset(args.batch_size)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\data_reader.py", line 801, in dataset
dataset = tf.compat.v1.data.make_one_shot_iterator(
Node: 'Input_Batch/IteratorGetNext'
Incompatible shapes at component 0: expected [?,3,1,31,1202,2] but got [1,3,1,31,1201,2].
[[{{node Input_Batch/IteratorGetNext}}]]
Original stack trace for 'Input_Batch/IteratorGetNext':
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 136, in <module>
main(args)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 129, in main
pred_fn(args, data_reader, log_dir=args.output_dir)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\predict.py", line 56, in pred_fn
data_batch = data_reader.dataset(args.batch_size)
File "D:\QuakeFlow\DeepDenoiser\deepdenoiser\data_reader.py", line 801, in dataset
dataset = tf.compat.v1.data.make_one_shot_iterator(
File "D:\miniconda3\lib\site-packages\tensorflow\python\data\ops\iterator_ops.py", line 440, in get_next
flat_ret = gen_dataset_ops.iterator_get_next(
File "D:\miniconda3\lib\site-packages\tensorflow\python\ops\gen_dataset_ops.py", line 3036, in iterator_get_next
_, _, _op, _outputs = _op_def_library._apply_op_helper(
File "D:\miniconda3\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 797, in _apply_op_helper
op = g._create_op_internal(op_type_name, inputs, dtypes=None,
File "D:\miniconda3\lib\site-packages\tensorflow\python\framework\ops.py", line 3800, in _create_op_internal
ret = Operation(
Metadata
Metadata
Assignees
Labels
No labels
