Добрый вечер, подскажите пожалуйста можно сделать флаг который бы указывал на какой из видеокарт будут инициализироваться слои и переменные. Грубо говоря после прохода функции энкод общие слои с функцией декод инициализировались на второй видеокарте и переменные тоже. Или для инициализации классов с функциями необходимо перед их запуском определить на какой видеокарте будут какие слои? То есть в процессе работы архитектуры общие слои для энкод и декод функций, классов не могут грузиться на другие видеокарты? Мне просто нужно разбить трансформер на две видеокарты.