search

USTC-Video-Understanding / I3D_Finetune

TensorFlow code for finetuning I3D model on UCF101.
144 stars 43 forks source link

When I train my data set，I got this problem. Why? num_class=2 #21

Closed Jiangchitang closed 5 years ago

Jiangchitang commented 5 years ago

InvalidArgumentError (see above for traceback): buffer_size must be greater than zero. [[Node: ShuffleDataset = ShuffleDataset[_class=["loc:@Iterator"], output_shapes=[[]], output_types=[DT_FLOAT], reshuffle_each_iteration=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](TensorSliceDataset, buffer_size, buffer_size, buffer_size)]]