diff options
| author | ericmarin <maarin.eric@gmail.com> | 2026-03-31 16:43:47 +0200 |
|---|---|---|
| committer | ericmarin <maarin.eric@gmail.com> | 2026-04-01 15:08:27 +0200 |
| commit | 81d4d604aa43660b732b3538734a52d509d7c5df (patch) | |
| tree | e0341280c3c3f10752aab7fccb2ddd5ed795c889 /examples/fashion_mnist/fashion_mnist.py | |
| parent | d1b25fbde6b01529fd1bcfdd5778b6cb378eb865 (diff) | |
| download | vein-81d4d604aa43660b732b3538734a52d509d7c5df.tar.gz vein-81d4d604aa43660b732b3538734a52d509d7c5df.zip | |
refactored examples
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: |