Skip to content

Commit f3f1ac8

Browse files
committed
fix examples
1 parent 081a2b8 commit f3f1ac8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/basic_tutorials/tutorial_mnist_simple.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def forward(self, x, foo=None):
6868

6969

7070
train_weights = MLP.trainable_weights
71-
optimizer = tl.optimizers.Momentum(0.05, 0.9)
71+
optimizer = tl.optimizers.Momentum(0.001, 0.9)
7272
metric = tl.metric.Accuracy()
7373
train_dataset = mnistdataset(data = X_train, label = y_train ,transform = transform)
7474
train_dataset = tl.dataflow.FromGenerator(train_dataset, output_types=[tl.float32, tl.int64], column_names=['data', 'label'])

0 commit comments

Comments
 (0)