Theano config directamente en script

Soy nuevo en Theano y me pregunto cómo configurar la configuración predeterminada directamente desde el script (sin configurar las variables de entorno). Por ejemplo, esta es una solución de trabajo ( fuente ):

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

Tengo la intención de encontrar la solución idéntica que ejecuta solo:

 $ python check1.py 

y los parámetros adicionales se establecen directamente en el propio script. Por ejemplo, de esta manera:

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

¿Es incluso posible? Leí la página de configuración que proporciona la información que me permite leer los valores ya establecidos (pero no configurarlos por mí mismo).

Cuando haces esto:

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

Todo lo que realmente está haciendo es configurar una variable de entorno antes de ejecutar el script de Python.

También puedes establecer variables de entorno en Python. Por ejemplo, la variable de entorno THEANO_FLAGS se puede configurar dentro de Python de la siguiente manera:

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

Tenga en cuenta que algunas variables de configuración de Theano no se pueden cambiar después de importar Theano, por lo que está bien:

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

Pero esto no funcionará como se esperaba:

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