Skip to content

Commit 445cca2

Browse files
committed
fixed cudnn problem
1 parent ec34da1 commit 445cca2

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

src/train_point_cloud.lua

-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@ else
4040
end
4141
local criterion = nn.ClassNLLCriterion()
4242
local criterion = criterion:cuda()
43-
criterion = cudnn.convert(criterion, cudnn)
4443

4544
local parameters, gradParameters = model:getParameters()
4645

0 commit comments

Comments
 (0)