由于我的 GPU 只支持 TensorFlow 2.x,我必须重写一个用 TensorFlow 1.x 编写的代码。
我使用了以下代码:https ://www.tensorflow.org/guide/upgrade
一切正常,直到我来到以下功能:
tf.contrib.training.bucket_by_sequence_length(input_length,
tensors,
batch_size,
bucket_boundaries,
num_threads=1,
capacity=32,
bucket_capacities=None,
shapes=None,
dynamic_pad=False,
allow_smaller_final_batch=False,
keep_input=True,
shared_name=None,
name=None)
不幸的是,因此我没有解决方案如何重写此代码。
我很高兴知道如何解决这个问题。