#tensorflow #tensorflow2.0 #automatic-differentiation #gradienttape
Вопрос:
Я начинаю записывать градиенты, tf.GradientTape(persistent=True) чтобы иметь возможность звонить gradient_tape.gradient() более одного раза. Поскольку это означает, что градиенты не освобождаются после вызова gradient() метода, должен ли я вызывать что-то вроде __exit__ в конце train_step ?