我是 Theano 新手,我想知道如何直接从脚本配置默认设置(无需设置 envir.variables)。例如。这是一个可行的解决方案(source http://deeplearning.net/software/theano/tutorial/using_gpu.html):
$ 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
有可能吗?我读了配置页 http://deeplearning.net/software/theano/library/config.html它提供的信息允许我读取已经设置的值(但不能自己设置)。
当你这样做时:
$ THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python check1.py
您实际上要做的就是在运行 Python 脚本之前设置一个环境变量。
您也可以在 Python 中设置环境变量。例如,THEANO_FLAGS
环境变量可以在 Python 中设置,如下所示:
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"
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)