Update number format and precision in training progress message

eic · Jan 12, 2024 · c4e1703 · c4e1703
1 parent 7398a0b
commit c4e1703
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/roman_pots/train_dense_neural_network.py b/benchmarks/roman_pots/train_dense_neural_network.py
@@ -77,7 +77,7 @@ def train_model(input_tensor, target_tensor, model, num_epochs, learning_rate):
 
     # Print progress
     if (epoch + 1) % 10 == 0:
-      print(f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item():.3e}')
+      print("Epoch "+str(epoch+1)+"/"+str(num_epochs)+", Loss: "+"{0:0.10f}".format(loss.item()))
 
   # Plot the loss values
   plt.plot(range(1, num_epochs+1), losses)