使用supervisor

时间:2018-03-05 23:24:50

标签: keras backend theano supervisor

如何在使用supervisor加载时动态设置python中的keras后端?当我使用supervisor加载python文件时,环境变量似乎会覆盖下面代码中的后端设置。

from keras import backend as K
import os
os.environ['KERAS_BACKEND'] = 'theano'
reload(K)

在环境变量中,Keras后端定义为“tensorflow”,因此脚本以'tensorflow'后端开头。上面的一行不起作用。

0 个答案:

没有答案