search

artyom-beilis / pytorch_dlprim

DLPrimitives/OpenCL out of tree backend for pytorch
http://blog.dlprimitives.org/
MIT License
264 stars 17 forks source link

inception_v3 gives incorrect prob values #3

Closed artyom-beilis closed 2 years ago

artyom-beilis commented 2 years ago

OpenCL

Accessing device #1:GeForce GTX 960 on NVIDIA CUDA
../dlprimitives/tests/samples/cat.ppm,281,tabby,-0.376586,-0.654800,-0.476176,-0.263846,-0.209125,-0.240114,0.732247,...
../dlprimitives/tests/samples/dog.ppm,207,golden retriever,0.329457,0.578268,0.332380,-0.336138,-0.431014,-0.998301,-0.334501,...
../dlprimitives/tests/samples/parrot.ppm,87,African grey,0.486966,0.501808,-0.003343,1.311452,-0.865675,-0.882673,-1.369494,...

CPU

../dlprimitives/tests/samples/cat.ppm,281,tabby,0.174380,-1.094309,-0.852133,0.360330,0.201213,0.684973,0.222629,...
../dlprimitives/tests/samples/dog.ppm,207,golden retriever,0.936222,1.471751,0.773317,0.216966,-0.954457,-2.085732,-1.479886,...
../dlprimitives/tests/samples/parrot.ppm,87,African grey,0.103325,0.242180,-0.443967,1.902003,-2.093624,-1.191285,-2.878913,...
artyom-beilis commented 2 years ago

Fixed