From c4e1703967c930975d0089ea4a029352339b7200 Mon Sep 17 00:00:00 2001 From: Sakib Rahman Date: Fri, 12 Jan 2024 12:46:42 -0500 Subject: [PATCH] Update number format and precision in training progress message --- benchmarks/roman_pots/train_dense_neural_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/roman_pots/train_dense_neural_network.py b/benchmarks/roman_pots/train_dense_neural_network.py index 3956b82d..56411dc9 100644 --- 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)