質問編集履歴

2

コードの内容を公式の例と同じようなものに変更

2018/05/16 03:34

投稿

elec
elec

スコア6

test CHANGED
File without changes
test CHANGED
@@ -14,19 +14,19 @@
14
14
 
15
15
  Traceback (most recent call last):
16
16
 
17
- File "test.py", line 10, in <module>
17
+ File "test.py", line 9, in <module>
18
18
 
19
- h = l(x)
19
+ y = l(x)
20
20
 
21
- File "/home/terao/.local/lib/python3.6/site-packages/chainer/links/connection/convolution_2d.py", line 171, in __call__
21
+ File "/home/user/.local/lib/python3.6/site-packages/chainer/links/connection/convolution_2d.py", line 171, in __call__
22
22
 
23
23
  groups=self.groups)
24
24
 
25
- File "/home/terao/.local/lib/python3.6/site-packages/chainer/functions/connection/convolution_2d.py", line 577, in convolution_2d
25
+ File "/home/user/.local/lib/python3.6/site-packages/chainer/functions/connection/convolution_2d.py", line 577, in convolution_2d
26
26
 
27
27
  y, = fnode.apply(args)
28
28
 
29
- File "/home/terao/.local/lib/python3.6/site-packages/chainer/function_node.py", line 235, in apply
29
+ File "/home/user/.local/lib/python3.6/site-packages/chainer/function_node.py", line 235, in apply
30
30
 
31
31
  ', '.join(str(type(x)) for x in in_data)))
32
32
 
@@ -48,23 +48,23 @@
48
48
 
49
49
  import chainer
50
50
 
51
- import chainer.links as L
51
+ import chainer.links as L
52
52
 
53
- import numpy as np
53
+ from chainer import cuda
54
54
 
55
55
 
56
56
 
57
- xp = chainer.backends.cuda.cupy
57
+ xp = cuda.cupy
58
58
 
59
- #xp = np
60
59
 
61
- l = L.Convolution2D(None, 5, 5, pad=2)
62
60
 
63
- x = chainer.Variable( xp.random.rand(100, 10, 150, 150).astype(xp.float32) )
61
+ x = xp.arange(1 * 3 * 10 * 10, dtype=xp.float32).reshape(1, 3, 10, 10)
64
62
 
65
- h = l(x)
63
+ l = L.Convolution2D(3, 7, 5)
66
64
 
65
+ y = l(x)
66
+
67
- print('finish')
67
+ print(y.shape)
68
68
 
69
69
  ```
70
70
 
@@ -74,7 +74,7 @@
74
74
 
75
75
 
76
76
 
77
- 入力をcupyでなくnumpyにしたとき(コメントを解除したとき)はうまくきました。
77
+ 入力をcupyでなくnumpyにしたときはうまくきました。
78
78
 
79
79
 
80
80
 

1

誤字

2018/05/16 03:34

投稿

elec
elec

スコア6

test CHANGED
@@ -1 +1 @@
1
- Convolution2DをGPU上で動かせません
1
+ ChainerのConvolution2DをGPU上で動かせません
test CHANGED
@@ -58,7 +58,7 @@
58
58
 
59
59
  #xp = np
60
60
 
61
- l = chainer.links.Convolution2D(None, 5, 5, pad=2)
61
+ l = L.Convolution2D(None, 5, 5, pad=2)
62
62
 
63
63
  x = chainer.Variable( xp.random.rand(100, 10, 150, 150).astype(xp.float32) )
64
64