Skip to content

Commit

Permalink
remove conv fallback
Browse files Browse the repository at this point in the history
  • Loading branch information
skaae committed Aug 19, 2015
1 parent 5f3ec72 commit cbafea6
Showing 1 changed file with 3 additions and 11 deletions.
14 changes: 3 additions & 11 deletions examples/spatial_transformer_network.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
"name": "stdout",
"output_type": "stream",
"text": [
"Using DNN layers\n"
""
]
},
{
Expand All @@ -34,16 +34,8 @@
"import lasagne\n",
"import theano\n",
"import theano.tensor as T\n",
"try:\n",
" from lasagne.layers import dnn # fails early if not available\n",
" conv = dnn.Conv2DDNNLayer\n",
" pool = dnn.MaxPool2DDNNLayer\n",
" print \"Using DNN layers\"\n",
"except:\n",
" conv = lasagne.layers.Conv2DLayer\n",
" pool = lasagne.layers.MaxPool2DLayer\n",
" print \"DNN not available, using standard (slower) conv and pool layers\"\n",
"\n",
"conv = lasagne.layers.Conv2DLayer\n",
"pool = lasagne.layers.MaxPool2DLayer\n",
"NUM_EPOCHS = 500\n",
"BATCH_SIZE = 256\n",
"LEARNING_RATE = 0.001\n",
Expand Down

0 comments on commit cbafea6

Please sign in to comment.