Skip to content

Commit 5f53158

Browse files
committed
New installation and minor fixes for theano 1.0.2
1 parent 7522834 commit 5f53158

File tree

3 files changed

+12
-8
lines changed

3 files changed

+12
-8
lines changed

.theanorc

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
[global]
22
floatX = float32
3-
device = gpu0
3+
device = cuda

README.md

+9-4
Original file line numberDiff line numberDiff line change
@@ -60,17 +60,22 @@ The package requires python3. You can follow the direction below to install virt
6060
git clone https://github.com/chrischoy/3D-R2N2.git
6161
```
6262

63-
- Setup virtual environment and install requirements and copy the theanorc file to the `$HOME` directory
63+
- Setup the anaconda virtual environment and installing requirements ([How to use anaconda](https://conda.io/docs/user-guide/install/index.html))
6464

6565
```
6666
cd 3D-R2N2
67-
pip install virtualenv
68-
virtualenv -p python3 --system-site-packages py3
69-
source py3/bin/activate
67+
conda create -n py3-theano python=3.6
68+
source activate py3-theano
69+
conda install pygpu
7070
pip install -r requirements.txt
71+
```
72+
- copy the theanorc file to the `$HOME` directory
73+
74+
```
7175
cp .theanorc ~/.theanorc
7276
```
7377

78+
7479
### Running demo.py
7580

7681
- Install meshlab (skip if you have another mesh viewer). If you skip this step, demo code will not visualize the final prediction.

lib/layers.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -319,12 +319,11 @@ def set_output(self):
319319
padded_input = self._prev_layer.output
320320
padded_input_shape = self._input_shape
321321

322-
conv_out = conv.conv2d(
322+
conv_out = tensor.nnet.conv2d(
323323
input=padded_input,
324324
filters=self.W.val,
325325
filter_shape=self._filter_shape,
326-
image_shape=np.asarray(
327-
padded_input_shape, dtype=np.int16),
326+
input_shape=padded_input_shape,
328327
border_mode='valid')
329328

330329
# add the bias term. Since the bias is a vector (1D array), we first

0 commit comments

Comments
 (0)