Конфигурация Theano непосредственно в script

Я новичок в Theano, и мне интересно, как настроить настройку по умолчанию непосредственно из script (без установки переменных envir.). Например. это рабочее решение (источник):

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

Я намереваюсь придумать то же решение, которое выполняется только:

$ python check1.py

а дополнительные параметры устанавливаются непосредственно в самом script. Например. так или иначе:

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

Возможно ли это? Я прочитал конфигурационную страницу, которая предоставляет информацию, которая позволяет мне читать уже установленные значения (но не устанавливать их самостоятельно).

Ответы

Ответ 1

Когда вы это сделаете:

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

Все, что вы на самом деле делаете, это установка переменной среды перед запуском Python script.

Вы также можете установить переменные среды в Python. Например, переменная среды THEANO_FLAGS может быть установлена ​​внутри Python следующим образом:

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

Обратите внимание, что некоторые переменные конфигурации Anano не могут быть изменены после импорта 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"