File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -104,7 +104,7 @@ MNIST Dataset
104104 Since now the data is in one variable, and a minibatch is defined as a
105105 slice of that variable, it comes more natural to define a minibatch by
106106 indicating its index and its size. In our setup the batch size stays constant
107- through out the execution of the code, therefore a function will actually
107+ throughout the execution of the code, therefore a function will actually
108108 require only the index to identify on which datapoints to work.
109109 The code below shows how to store your data and how to
110110 access a minibatch:
@@ -141,8 +141,8 @@ MNIST Dataset
141141
142142 # accessing the third minibatch of the training set
143143
144- data = train_set_x[2 * 500 : 3 * 500 ]
145- label = train_set_y[2 * 500 : 3 * 500 ]
144+ data = train_set_x[2 * batch_size : 3 * batch_size ]
145+ label = train_set_y[2 * batch_size : 3 * batch_size ]
146146
147147
148148The data has to be stored as floats on the GPU ( the right
You can’t perform that action at this time.
0 commit comments