使用 tensorflow 2.4.1

如何解决使用 tensorflow 2.4.1

我将尝试使用网站 https://github.com/TannerGilbert/Tensorflow-Object-Detection-API-Train-Model 中提到的 Tensorflow 对象检测 API 训练自定义对象检测模型。正如网站中提到的，我遵循了所有内容并代替了网站中使用的数据集，我使用了自己的数据集，其尺寸为 750 像素 x 750 像素，数据类型为“uint8”。

同样代替教程中使用的模型（efficientdet_d0_coco17_tpu-32.tar），我使用了“efficientdet_d2_coco17_tpu-32.tar”模型。

但是在我们必须训练模型的第 6 步中，我收到了错误，如此 https://drive.google.com/file/d/13vBn1pkkK3xHL01cFZuWrSdOh2n-kgwE/view?usp=sharing 所示。

在互联网上搜索解决方案后，我尝试将以下代码合并到我的 model_main_tf2.py 中，但问题仍然存在：

try:
    tf_gpus = tf.config.list_physical_devices('GPU')
    for gpu in tf_gpus:
        tf.config.experimental.set_memory_growth(gpu,True)
except:
    pass

现在我收到此错误 https://drive.google.com/file/d/19QMp_vVbV2kk3Rvix7Kp5ZAyzq-cGGwQ/view?usp=sharing

谁能告诉我如何解决这个问题。