diff options
Diffstat (limited to 'examples/fashion_mnist/fashion_mnist.py')
| -rw-r--r-- | examples/fashion_mnist/fashion_mnist.py | 2 |
1 files changed, 1 insertions, 1 deletions
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: |