環(huán)境

• Ubuntu 18.04
• Tensorflow-gpu 2.4

報(bào)錯(cuò)

Traceback (most recent call last):File "nn/cnn.py", line 134, in <module>run()File "nn/cnn.py", line 128, in runtrainer.train(train_dataloader, valid_dataloader, **fit_params)File "/home/teletraan/baseline/competition/earth/code/nn/trainer_tf.py", line 28, in trainvalid_epoch_loss, valid_scores = self.valid_loop(valid_dataset)File "/home/teletraan/baseline/competition/earth/code/nn/trainer_tf.py", line 81, in valid_loopy_valid_preds = tf.concat(y_valid_preds, axis=0)File "/home/teletraan/.pyenv/versions/comp/lib/python3.7/site-packages/tensorflow/python/util/dispatch.py", line 201, in wrapperreturn target(*args, **kwargs)File "/home/teletraan/.pyenv/versions/comp/lib/python3.7/site-packages/tensorflow/python/ops/array_ops.py", line 1654, in concatreturn gen_array_ops.concat_v2(values=values, axis=axis, name=name)File "/home/teletraan/.pyenv/versions/comp/lib/python3.7/site-packages/tensorflow/python/ops/gen_array_ops.py", line 1207, in concat_v2_ops.raise_from_not_ok_status(e, name)File "/home/teletraan/.pyenv/versions/comp/lib/python3.7/site-packages/tensorflow/python/framework/ops.py", line 6843, in raise_from_not_ok_statussix.raise_from(core._status_to_exception(e.code, message), None)File "<string>", line 3, in raise_from tensorflow.python.framework.errors_impl.InvalidArgumentError: OpKernel 'ConcatV2' has constraint on attr 'T' not in NodeDef '[N=0, Tidx=DT_INT32]', KernelDef: 'op: "ConcatV2" device_type: "GPU" constraint { name: "T" allowed_values { list { type: DT_INT32 } } } host_memory_arg: "values" host_memory_arg: "axis" host_memory_arg: "output"' [Op:ConcatV2] name: concat

問(wèn)題

由于y_valid是空，導(dǎo)致concat操作報(bào)錯(cuò)。
我這里的根本原因是batch_size設(shè)置太大，大于樣本數(shù)了，所以出現(xiàn)以上錯(cuò)誤。調(diào)低batch_size即可

總結(jié)

以上是生活随笔為你收集整理的Debug Tensorflow: tensorflow.python.framework.errors_impl.InvalidArgumentError: OpKernel ‘ConcatV2‘的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。