Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def train(loss, model, opt, original):
- with tf.GradientTape() as tape:
- gradients = tape.gradient(loss(model, original), model.trainable_variables)
- gradient_variables = zip(gradients, model.trainable_variables)
- opt.apply_gradients(gradient_variables)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement