Skip to content

Commit 5167d62

Browse files
authored
increase mnist training epochs (aws#87)
1 parent 3c17cba commit 5167d62

File tree

1 file changed

+1
-1
lines changed
  • sagemaker-python-sdk/mxnet_mnist

1 file changed

+1
-1
lines changed

sagemaker-python-sdk/mxnet_mnist/mnist.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def train(channel_input_dirs, hyperparameters, hosts, num_cpus, num_gpus, **kwar
5353
optimizer_params={'learning_rate': float(hyperparameters.get("learning_rate", 0.1))},
5454
eval_metric='acc',
5555
batch_end_callback=mx.callback.Speedometer(batch_size, 100),
56-
num_epoch=10)
56+
num_epoch=25)
5757
return mlp_model
5858

5959

0 commit comments

Comments
 (0)