keras多GPU模型和单GPU权模型并不相同,权重也不相同,从多GPU权重提取单GPU权重的操作示例为:
model = ResNextImageNet(blah..)
p_model = multi_gpu_model(model)
p_model.load_weights(bst) # load multi-gpu model weights
old_model = p_model.layers[-2] #get single GPU model weights
# it's necessary to save the model before use this single GPU model
old_model.save(save_path)
参考: