Applying the tensorflow tutorial ?
培训――步骤功能如下:
@tf.function(input_signature=train_step_signature)
def train_step(inp, tar):
tar_inp = tar[:, :-1]
tar_real = tar[:, 1:]
with tf.GradientTape() as tape:
predictions, _ = transformer([inp, tar_inp],
training = True)
loss = loss_function(tar_real, predictions)
gradients = tape.gradient(loss, transformer.trainable_variables)
optimizer.apply_gradients(zip(gradients, transformer.trainable_variables))
train_loss(loss)
train_accuracy(accuracy_function(tar_real, predictions))
我们可以看到这一点。 Gra(GradientTape)的定义是带有雕像的磁带。 这些工作,但我不理解如何在发言之外打电话录音带。
gradients = tape.gradient(loss, transformer.trainable_variables)
是否应在发言结束时关闭录音带? 如果你能够回答我的好奇,非常感谢你。
我从辅导员处执行该守则,并且照样工作。