-
smallworld-network-wupeng
我生成了两个tfrecord ,但是运行train_smiple.py的时候还是报了错:
Traceback (most recent call last): File "train_simple.py", line 120, in <module> train() File "train_simple.py", line 49, in train for data in train_dataset.take(2): File "/home/wupeng/anaconda3/envs/tf2/lib/python3.7/site-packages/tensorflow/python/data/ops/iterator_ops.py", line 736, in __next__ return self.next() File "/home/wupeng/anaconda3/envs/tf2/lib/python3.7/site-packages/tensorflow/python/data/ops/iterator_ops.py", line 772, in next return self._next_internal() File "/home/wupeng/anaconda3/envs/tf2/lib/python3.7/site-packages/tensorflow/python/data/ops/iterator_ops.py", line 764, in _next_internal return structure.from_compatible_tensor_list(self._element_spec, ret) File "/home/wupeng/anaconda3/envs/tf2/lib/python3.7/contextlib.py", line 130, in __exit__ self.gen.throw(type, value, traceback) File "/home/wupeng/anaconda3/envs/tf2/lib/python3.7/site-packages/tensorflow/python/eager/context.py", line 2105, in execution_mode executor_new.wait() File "/home/wupeng/anaconda3/envs/tf2/lib/python3.7/site-packages/tensorflow/python/eager/executor.py", line 67, in wait pywrap_tfe.TFE_ExecutorWaitForAllPendingNodes(self._handle) tensorflow.python.framework.errors_impl.InvalidArgumentError: Input to reshape is a tensor with 7830 values, but the requested shape has 4096 [[{{node Reshape}}]]