We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 081a2b8 commit f3f1ac8Copy full SHA for f3f1ac8
examples/basic_tutorials/tutorial_mnist_simple.py
@@ -68,7 +68,7 @@ def forward(self, x, foo=None):
68
69
70
train_weights = MLP.trainable_weights
71
-optimizer = tl.optimizers.Momentum(0.05, 0.9)
+optimizer = tl.optimizers.Momentum(0.001, 0.9)
72
metric = tl.metric.Accuracy()
73
train_dataset = mnistdataset(data = X_train, label = y_train ,transform = transform)
74
train_dataset = tl.dataflow.FromGenerator(train_dataset, output_types=[tl.float32, tl.int64], column_names=['data', 'label'])
0 commit comments