Theano config bezpośrednio w skrypcie

Question

Jestem nowym użytkownikiem Theano i zastanawiam się, jak skonfigurować domyślne ustawienie bezpośrednio ze skryptu (bez ustawiania zmiennych środowiskowych). Na przykład. Jest to roztwór roboczy (source):

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

mam zamiar wymyślić identycznego rozwiązania, które jest wykonywane tylko przez:

$ python check1.py 

i dodatkowe parametry są ustawione bezpośrednio w samym skrypcie. Na przykład. w pewnym sensie:

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

Czy to możliwe? Przeczytałem config page, który dostarcza informacji, które pozwalają mi odczytać już ustawione wartości (ale nie ustawić ich samodzielnie).

Answer 1

Kiedy to zrobić:

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

All jesteś rzeczywiście robi jest ustawienie zmiennej środowiskowej przed uruchomieniem skryptu Pythona.

Możesz ustawić zmienne środowiskowe także w Pythonie. Na przykład, zmienna THEANO_FLAGS środowisko może być ustawiony wewnątrz Python tak:

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

pamiętać, że niektóre zmienne konfiguracyjne Theano nie może być zmieniony po zaimportowaniu Theano tak to jest w porządku:

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

Ale to nie będzie działać zgodnie z oczekiwaniami:

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