我试图为一些验证测试按顺序加载多个模型,但是使用K.clear_session()
会导致Python在Windows上崩溃。你知道吗
"Python is not working/resonding" Windows error.
如果我不使用K.clear_session()
,我会遇到内存错误,
我假设图继续堆叠,每个模型
逐步加载需要更长的时间。你知道吗
我正在尝试下面关于如何使用K.clear_session()
的建议,但是Python不断崩溃(我没有得到错误,只是崩溃)
Extremely slow model load with keras
这是我的密码:
>>> print(keras.__version__)
2.2.4
>>> import tensorflow as tf
>>> tf.__version__
'1.8.0'
>>>
Windows 7
config = tf.ConfigProto()
# Don't pre-allocate memory; allocate as-needed
config.gpu_options.allow_growth = True
# Only allow a total of half the GPU memory to be allocated
config.gpu_options.per_process_gpu_memory_fraction = 0.8
# Create a session with the above options specified.
k.tensorflow_backend.set_session(tf.Session(config=config))
def evaluate_models_2(models_path_dir,):
models_paths = [os.path.join(models_path_dir, model) for model in os.listdir(models_path_dir) if model.endswith(".hdf5")]
models_pairs = get_model_key(models_paths, model_keys, MODEL_KEY)
valid_gen = double_valid_generator()
#global model_paths
for num,model_pair in enumerate(models_pairs):
model_path,model_key = model_pair
print(num, ": ", model_path)
evaluate_validation_data_2(model_path, model_key, generator = valid_gen)
def evaluate_validation_data_2(model_path,model_key, generator = None, show_mistakes = False):
if generator == None:
valid_gen = double_valid_generator()
else:
valid_gen = generator
#load data generator
print("about to load model")
model = load_model_max_layer(model_path)
#evaluate and then clear session
print(model.evaluate_generator(valid_gen, steps =10))
k.clear_session()
我似乎没有其他的Python会话在运行。你知道吗
如果我删除K.clear\u session(),python不会崩溃 然而,如前所述,模型需要越来越长的时间来加载,我有大约200个模型要经历。你知道吗
目前没有回答
相关问题 更多 >
编程相关推荐