File tree 3 files changed +12
-8
lines changed
3 files changed +12
-8
lines changed Original file line number Diff line number Diff line change 1
1
[global]
2
2
floatX = float32
3
- device = gpu0
3
+ device = cuda
Original file line number Diff line number Diff line change @@ -60,17 +60,22 @@ The package requires python3. You can follow the direction below to install virt
60
60
git clone https://github.com/chrischoy/3D-R2N2.git
61
61
```
62
62
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 ) )
64
64
65
65
```
66
66
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
70
70
pip install -r requirements.txt
71
+ ```
72
+ - copy the theanorc file to the ` $HOME ` directory
73
+
74
+ ```
71
75
cp .theanorc ~/.theanorc
72
76
```
73
77
78
+
74
79
### Running demo.py
75
80
76
81
- Install meshlab (skip if you have another mesh viewer). If you skip this step, demo code will not visualize the final prediction.
Original file line number Diff line number Diff line change @@ -319,12 +319,11 @@ def set_output(self):
319
319
padded_input = self ._prev_layer .output
320
320
padded_input_shape = self ._input_shape
321
321
322
- conv_out = conv .conv2d (
322
+ conv_out = tensor . nnet .conv2d (
323
323
input = padded_input ,
324
324
filters = self .W .val ,
325
325
filter_shape = self ._filter_shape ,
326
- image_shape = np .asarray (
327
- padded_input_shape , dtype = np .int16 ),
326
+ input_shape = padded_input_shape ,
328
327
border_mode = 'valid' )
329
328
330
329
# add the bias term. Since the bias is a vector (1D array), we first
You can’t perform that action at this time.
0 commit comments