无法在 python 中为segmentation_models库选择框架



我正在尝试使用python中的segmentation_models(sm(库构建ResNet34模型。sm库在导入时默认使用 keras 框架,但我正在使用tf.keras来构建用于训练和测试的数据集。

文档说,为了更改默认框架,我应该在导入之前使用环境变量SM_FRAMEWORK=tf.keras(我尝试过但没有工作(或使用方法set_framework设置它(它没有显示在建议中/它说当我尝试执行它时它不存在(。

还有其他方法可以克服这个问题吗?

我在colab上训练了模型,并使用:"%env SM_FRAMEWORK=tf.keras"将环境设置为tf.keras,它运行良好。

最新更新