Problem when running: Check failed: size >= 0 (-1092442872 vs. 0)

[xiaojxkevin]

Dear author, I am trying to run this model on my own dataset for tests, and I have followed the instructions in Readme.md (except that I use the local data). However, After I run python Train.py --dataset 3, I receive the following results: Epoch: 0 of 1 2023-07-25 15:55:24.120018: I tensorflow/stream_executor/dso_loader.cc:152] successfully opened CUDA library libcublas.so.10.0 locally 2023-07-25 15:55:24.319214: F tensorflow/core/framework/tensor_shape.cc:241] Check failed: size >= 0 (-1092442872 vs. 0) I am wondering what may cause this and how can I fix it. Thanks!