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

shufflenet_v2_x1_0 calculations aren't correct: #2

Closed artyom-beilis closed 2 years ago

artyom-beilis commented 2 years ago

Output:

python ../dlprimitives/tools/validate_network.py --model shufflenet_v2_x1_0 --device opencl:1 ../dlprimitives/tests/samples/*.ppm  
Accessing device #1:GeForce GTX 960 on NVIDIA CUDA
../dlprimitives/tests/samples/cat.ppm,10,brambling,21.260544,12.555037,10.019163,31.069736,8.794845,8.340912,14.425335,...
../dlprimitives/tests/samples/dog.ppm,10,brambling,24.036289,12.404519,12.064783,34.494076,8.115385,11.757505,15.078144,...
../dlprimitives/tests/samples/parrot.ppm,463,bucket,26.767506,8.116230,7.385152,26.629230,8.159229,13.571995,12.405827,...

Expected:

$ python ../dlprimitives/tools/validate_network.py --model shufflenet_v2_x1_0 --device cpu ../dlprimitives/tests/samples/*.ppm  
../dlprimitives/tests/samples/cat.ppm,281,tabby,-3.539900,-1.827396,-1.215353,-2.049888,0.358607,0.946634,0.864965,...
../dlprimitives/tests/samples/dog.ppm,207,golden retriever,0.087652,0.828827,1.272321,1.876501,1.484513,-0.203963,0.274946,...
../dlprimitives/tests/samples/parrot.ppm,87,African grey,5.516771,-0.203844,-1.753324,-0.808835,-0.437290,-1.787125,-1.651563,...
artyom-beilis commented 2 years ago

Fixed in https://github.com/artyom-beilis/pytorch_dlprim/commit/8a4dc83918a6b9e81c02d3a95c0527d7ac90688e