Skip to content

Commit 4544ace

Browse files
author
Ragav Venkatesan
authored
notes on gpu/cpu changes.
1 parent a600743 commit 4544ace

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

advanced_functionality/mxnet_mnist_byom/mxnet_mnist.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@
9898
"source": [
9999
"### Training\n",
100100
"\n",
101-
"It is time to train the network. Since we are training the network locally, we can make use of mxnet training tools. The training method is also in the accompanying [mnist.py](mnist.py) file. The method is shown below. \n",
101+
"It is time to train the network. Since we are training the network locally, we can make use of mxnet training tools. The training method is also in the accompanying [mnist.py](mnist.py) file. The notebook assumes that this instance is a `p2.xlarge`. If running this in a non-GPU notebook instance, please adjust num_gpus=0 and num_cpu=1 The method is shown below. \n",
102102
"\n",
103103
"```python \n",
104104
"def train(data, hyperparameters= {'learning_rate': 0.11}, num_cpus=0, num_gpus =1 , **kwargs):\n",

0 commit comments

Comments
 (0)