From 81d4d604aa43660b732b3538734a52d509d7c5df Mon Sep 17 00:00:00 2001 From: ericmarin Date: Tue, 31 Mar 2026 16:43:47 +0200 Subject: refactored examples --- examples/fashion_mnist/fashion_mnist.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'examples/fashion_mnist') diff --git a/examples/fashion_mnist/fashion_mnist.py b/examples/fashion_mnist/fashion_mnist.py index 3514448..680f4eb 100644 --- a/examples/fashion_mnist/fashion_mnist.py +++ b/examples/fashion_mnist/fashion_mnist.py @@ -23,7 +23,7 @@ def train_model(name: str, dim): loss_fn = nn.CrossEntropyLoss() optimizer = torch.optim.Adam(net.parameters(), lr=1e-4) - print(f"Training {name}...") + print(f"Training {name} ({dim} neurons)...") for epoch in range(10): global loss for data in trainloader: -- cgit v1.2.3