Theano直接在脚本

Question

我是Theano的新手，我想知道如何直接从脚本配置默认设置（不设置envir。变量）。例如。这是一个有效的解决方案（source）：

$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python check1.py

我打算提出仅由以下方式执行的相同解决方案：

$ python check1.py

，其他参数直接在脚本中设置。例如。不知何故这样：

import theano
theano.set('mode', 'FAST_RUN')
theano.set('device', 'gpu')
theano.set('floatX', 'float32')
# rest of the script

甚至可能吗？我读了config page，它提供了允许我读取已经设置的值的信息（但不是我自己设置的）。

Answer 1

执行此操作时：

$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python check1.py

您所做的只是在运行Python脚本之前设置环境变量。

您也可以在Python中设置环境变量。例如，可以在Python中设置THEANO_FLAGS环境变量，如下所示：

import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"

请注意，导入Theano后无法更改某些Theano配置变量，所以这很好：

import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"
import theano

但这不会按预期工作：

import theano
import os
os.environ["THEANO_FLAGS"] = "mode=FAST_RUN,device=gpu,floatX=float32"