alihan/TensorFlow-2.x-Tutorials
1
0
Fork 0
mirror of https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git synced 2021-05-12 18:32:23 +03:00
Code Issues Projects Releases Wiki Activity

add lesson 1

Browse Source
This commit is contained in:
dragen1860
2019-01-31 16:04:38 +11:00
parent 90be70faac
commit 4f6ea60771
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lesson1-Introduction/main.py
View File

@@ -39,7 +39,9 @@ def train_one_step(model, optimizer, x, y):
logits = model(x)
loss = compute_loss(logits, y)
# compute gradient
grads = tape.gradient(loss, model.trainable_variables)
# update to weights
optimizer.apply_gradients(zip(grads, model.trainable_variables))
accuracy = compute_accuracy(logits, y)
@@ -77,7 +79,7 @@ def main():
for epoch in range(20):
loss, accuracy = train(epoch, model, optimizer)
print('Final epoch', epoch, ': loss', loss, '; accuracy', accuracy)
print('Final epoch', epoch, ': loss', loss.numpy(), '; accuracy', accuracy.numpy())
if __name__ == '__main__':