Hello! I just updated everything and I’m running the toy example from the Quick Start and I got this error:
`Caused by op u'seq2seq/encoder/w_embs/Initializer/random_uniform/RandomUniform', defined at:
File "/usr/local/bin/onmt-main", line 11, in <module> sys.exit(main()) File "/usr/local/lib/python2.7/dist-packages/opennmt/bin/main.py", line 150, in main runner.train_and_evaluate() File "/usr/local/lib/python2.7/dist-packages/opennmt/runner.py", line 167, in train_and_evaluate tf.estimator.train_and_evaluate(self._estimator, train_spec, eval_spec) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/training.py", line 447, in train_and_evaluate return executor.run() File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/training.py", line 531, in run return self.run_local() File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/training.py", line 669, in run_local hooks=train_hooks) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator.py", line 366, in train loss = self._train_model(input_fn, hooks, saving_listeners) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator.py", line 1119, in _train_model return self._train_model_default(input_fn, hooks, saving_listeners) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator.py", line 1132, in _train_model_default features, labels, model_fn_lib.ModeKeys.TRAIN, self.config) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator.py", line 1107, in _call_model_fn model_fn_results = self._model_fn(features=features, **kwargs) File "/usr/local/lib/python2.7/dist-packages/opennmt/models/model.py", line 103, in _model_fn _loss_op, features_shards, labels_shards, params, mode, config) File "/usr/local/lib/python2.7/dist-packages/opennmt/utils/parallel.py", line 148, in __call__ outputs.append(funs[i](*args[i], **kwargs[i])) File "/usr/local/lib/python2.7/dist-packages/opennmt/models/model.py", line 66, in _loss_op logits, _ = self._build(features, labels, params, mode, config=config) File "/usr/local/lib/python2.7/dist-packages/opennmt/models/sequence_to_sequence.py", line 144, in _build scope=source_input_scope)(features) File "/usr/local/lib/python2.7/dist-packages/opennmt/models/sequence_to_sequence.py", line 50, in _scoped_embedding_fn return embedding_fn(ids) File "/usr/local/lib/python2.7/dist-packages/opennmt/models/sequence_to_sequence.py", line 143, in <lambda> lambda ids: self.source_inputter.transform_data(ids, mode=mode, log_dir=log_dir), File "/usr/local/lib/python2.7/dist-packages/opennmt/inputters/inputter.py", line 194, in transform_data inputs = self._transform_data(data, mode) File "/usr/local/lib/python2.7/dist-packages/opennmt/inputters/text_inputter.py", line 361, in _transform_data return self.transform(data["ids"], mode) File "/usr/local/lib/python2.7/dist-packages/opennmt/inputters/text_inputter.py", line 388, in transform trainable=self.trainable) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 1328, in get_variable constraint=constraint) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 1090, in get_variable constraint=constraint) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 435, in get_variable constraint=constraint) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 404, in _true_getter use_resource=use_resource, constraint=constraint) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 796, in _get_single_variable use_resource=use_resource) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 2234, in variable use_resource=use_resource) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 2224, in <lambda> previous_getter = lambda **kwargs: default_variable_creator(None, **kwargs) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 2207, in default_variable_creator constraint=constraint) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variables.py", line 259, in __init__ constraint=constraint) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variables.py", line 368, in _init_from_args initial_value(), name="initial_value", dtype=dtype) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/variable_scope.py", line 780, in <lambda> shape.as_list(), dtype=dtype, partition_info=partition_info) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/init_ops.py", line 476, in __call__ shape, -limit, limit, dtype, seed=self.seed) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/random_ops.py", line 242, in random_uniform rnd = gen_random_ops.random_uniform(shape, dtype, seed=seed1, seed2=seed2) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/gen_random_ops.py", line 674, in random_uniform name=name) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper op_def=op_def) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 3414, in create_op op_def=op_def) File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/framework/ops.py", line 1740, in __init__ self._traceback = self._graph._extract_stack() # pylint: disable=protected-access
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[24999,512] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[Node: seq2seq/encoder/w_embs/Initializer/random_uniform/RandomUniform = RandomUniform[T=DT_INT32, _class=["loc:@seq2seq/encoder/w_embs/Assign"], dtype=DT_FLOAT, seed=0, seed2=0, _device="/job:localhost/replica:0/task:0/device:GPU:0"](seq2seq/encoder/w_embs/Initializer/random_uniform/shape)]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
Any ideas??
Thanks in advance!!