Я реализую пользовательский слой с API Keras (работаю с TF2.0-beta
).Я хочу использовать номер эпохи в моих вычислениях, чтобы затратить параметр с течением времени (значение - в методе call()
).
Я привык к tf.get_global_step()
, но понимаю, что TF устарел во всех глобальныхобъемы, и, безусловно, по уважительной причине.
Если бы у меня был экземпляр модели, я мог бы использовать model.optimizer.iterations
, но я не уверен, как получить экземпляр моей родительской модели при реализации Layer.
ВыполнитьУ меня есть какой-либо способ сделать это, или единственный способ - позволить слою выставить обратный вызов, который обновит параметр, который я хочу убить?Другие идеи?В идеале это то, что не заставит пользователя уровня знать об этой внутренней детали (поэтому мне не нравится подход Callback - пользователь должен добавить их в модель).