facebookarchive / C3D

C3D is a modified version of BVLC caffe to support 3D ConvNets.
Other
1.16k stars 506 forks source link

different results produced by CPU and GPU mode #420

Closed mmchen88 closed 5 years ago

mmchen88 commented 5 years ago

I use the same UCF101 model, same data ,but get different results when using CPU mode and GPU mode.

dutran commented 5 years ago

how much different? is it about machine epsilon? CPU and GPU may have different floating point precison.

mmchen88 commented 5 years ago

I found the reason, CudnnReluLayer in GPU mode only process 4D tensors. When the tensor is 5D, the code must be modified.