Можете ли вы обновить параметры модуля из nn.compact этого модуля? (самоизменяющиеся сети)

Я новичок в льне, и мне было интересно, как правильно получить такое поведение:

      param = f.init(key,x) 
new_param, y = f.apply(param,x) 

Где f — экземпляр nn.module.
Где f может пройти через несколько операций, чтобы получить new_param, и что эти операции могут полагаться на промежуточный параметр для получения своего вывода.

Итак, в принципе, есть ли способ получить доступ и обновить параметры, предоставленные экземпляру nn.module, из__call__, не теряя при этом функционального свойства, поэтому все это можно обернуть преобразованием функции градиента.

0 ответов

Другие вопросы по тегам