У меня есть 2 CNN:
кодер CNN A
декодер CNN B
Я объединяю их в A1-> B-> A2, где A2 и A1 идентична A и имеет одни и те же переменные.
И, наконец, я добавляю потери на A2
во время тренировки, я хочу игнорировать градиенты от A2 (относительно A2 исправлено). Просто тренируйте градиенты от B и A1
Поскольку A1 и A2 используют одни и те же переменные, я не могу указать optimizer.minimize (var_list = [...]).
Я знаю, что могу явно сделать переменные, используя var1 = tf.Variable (...) и добавив var2 = tf.stop_gradient (var1) в A2. Но это слишком неудобно, так как я люблю использовать tf.layers
Есть ли простой способ?