Theano Config direkt im Skript

Question

Ich bin neu in Theano und ich frage mich, wie Sie die Standardeinstellung direkt aus dem Skript konfigurieren (ohne Einstellung von Umgebungsvariablen). Z.B. dies ist eine Arbeitslösung (source):

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

Ich beabsichtige, mit der identischen Lösung zu entwickeln, die nur ausgeführt wird:

$ python check1.py 

und die zusätzlichen Parameter werden direkt im Skript selbst festgelegt. Z.B. irgendwie wie folgt:

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

Ist es überhaupt möglich? Ich lese die config page, die die Informationen zur Verfügung stellt, die es mir erlauben, die bereits gesetzten Werte zu lesen (aber nicht, um sie selbst zu setzen).

Answer 1

Wenn Sie dies tun:

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

Alles, was Sie tatsächlich eine Umgebungsvariable zu tun setzt, bevor das Python-Skript ausgeführt wird.

Sie können auch Umgebungsvariablen in Python festlegen. Zum Beispiel kann die THEANO_FLAGS Umgebungsvariable innerhalb Python wie folgt festgelegt werden:

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

Beachten Sie, dass einige Theano Konfigurationsvariablen können nicht nach dem Import Theano verändert werden, so ist dies in Ordnung:

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

aber dies wird nicht funktionieren wie erwartet:

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