Merge pull request #625 from mselensky/gh-pages

tab indent typo fix for example TensorFlow script
NREL · May 14, 2024 · c9f9140 · c9f9140
2 parents 8cda9fa + 3da2039
commit c9f9140
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/Friendly_User/gpu_containers.md b/docs/Friendly_User/gpu_containers.md
@@ -152,8 +152,8 @@ Now we are ready to run the Tensorflow container. We will run the script below,
         gradients = tape.gradient(loss, model.trainable_variables)
         optimizer.apply_gradients(zip(gradients, model.trainable_variables))
 
-    train_loss(loss)
-    train_accuracy(labels, predictions)
+        train_loss(loss)
+        train_accuracy(labels, predictions)
 
     # test the model
     @tf.function
@@ -163,8 +163,8 @@ Now we are ready to run the Tensorflow container. We will run the script below,
         predictions = model(images, training=False)
         t_loss = loss_object(labels, predictions)
 
-    test_loss(t_loss)
-    test_accuracy(labels, predictions)
+        test_loss(t_loss)
+        test_accuracy(labels, predictions)
 
 
     t0 = time.time()