如何在gpu上加载keras模型(在cpu上训练)

时间:2020-02-19 17:03:50

标签: keras model neural-network load

假设我已经在CPU上使用Keras训练了神经网络并相应地保存了权重。现在,如果我要在GPU上执行相同的代码,那么当我加载权重时,是否需要进行任何更改,例如传递执行该设备时

model.load_weights('x.h5')

0 个答案:

没有答案