fix for train
Browse files- mnist.pth +0 -0
- train_conv.py +2 -1
mnist.pth
CHANGED
|
Binary files a/mnist.pth and b/mnist.pth differ
|
|
|
train_conv.py
CHANGED
|
@@ -3,12 +3,13 @@ import torch.nn as nn
|
|
| 3 |
import torch.optim as optim
|
| 4 |
from torchvision import datasets, transforms
|
| 5 |
from models import NetConv
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
# Training settings
|
| 9 |
batch_size = 64
|
| 10 |
test_batch_size = 1000
|
| 11 |
-
epochs =
|
| 12 |
lr = 0.01
|
| 13 |
momentum = 0.5
|
| 14 |
seed = 1
|
|
|
|
| 3 |
import torch.optim as optim
|
| 4 |
from torchvision import datasets, transforms
|
| 5 |
from models import NetConv
|
| 6 |
+
import torch.nn.functional as F
|
| 7 |
|
| 8 |
|
| 9 |
# Training settings
|
| 10 |
batch_size = 64
|
| 11 |
test_batch_size = 1000
|
| 12 |
+
epochs = 20
|
| 13 |
lr = 0.01
|
| 14 |
momentum = 0.5
|
| 15 |
seed = 1
|