• 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}}]]
    

    发布在 社区求助区(SOS!!) 阅读更多

与 神力AI社区 的连接断开,我们正在尝试重连,请耐心等待