web-dev-qa-db-ja.com

スクリプトで直接Theanoconfig

私はTheanoを初めて使用しますが、デフォルト設定をスクリプトから直接構成する方法を知りたいです（環境変数を設定せずに）。例えば。これは実用的な解決策です（ ソース ）：

$ 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 を読みました。これは、すでに設定されている値を読み取るための情報を提供します（ただし、自分で設定することはできません）。

pythontheano
13
petrbel

これを行うとき：

$ 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"
27
Daniel Renshaw