我在GPU上训练了一个模型,现在我正试图在CPU上对其进行评估(GPU正用于不同的训练运行)。但是,当我尝试使用以下命令加载它时: with tf.device('/cpu:0'):
model = tf.keras.models.load_model('models/resnet50_untrained.h5', compile=False) 我要
我想知道设置创建/训练模型的设备以优化资源使用的正确方法是什么,以便使用Keras在TensorFlow中进行快速培训?我有1个CPU和2个GPU可供我使用。最初,我使用tf.device上下文来创建模型并仅在CPU上进行培训,但后来我在TensorFlow文档中看到了,它们建议在CPU上显式实例化模型:
# Instantiate t
我试图将py文件编译为二进制文件,该二进制文件简单地从json文件读取模型并通过导入的模型预测输出。现在的问题是,当我试图通过pyinstaller编译程序时,生成的文件大约为290 it,因为它试图编译整个tensorflow包和它不想要的依赖项。更不用说,在尝试提取文件时,启动非常慢。import os
from tensorflow.keras.preprocessing.image import ImageDataGenera