完成实验2

2023-10-24 20:15:43 +08:00
parent 7e8fea0a3b
commit d76db395de
10 changed files with 739 additions and 161 deletions
--- a/Lab2/code/2.3.py
+++ b/Lab2/code/2.3.py
@@ -99,7 +99,7 @@ if __name__ == "__main__":
        avg_epoch_acc = total_epoch_acc / len(test_mnist_dataset)
        print(
            f"Epoch [{epoch + 1}/{num_epochs}],",
-            f"Train Loss: {total_epoch_loss},",
+            f"Train Loss: {total_epoch_loss:.10f},",
            f"Used Time: {train_time * 1000:.3f}ms,",
            f"Test Acc: {avg_epoch_acc * 100:.3f}%,",
            f"Used Time: {test_time * 1000:.3f}ms",