burn
burn copied to clipboard
Candle backend crashes web app in image-classification-web
Describe the bug Candle backend crashes web app in image-classification-web.
To Reproduce
- cd
examples/image-classification-web - run
./build-for-web.sh - run
./run-server.sh - open the browser with the served link
- change backed to Candle
- select image.
- open debug console and you'll notice error
Expected behavior The image is classified without a crash.
Screenshots
The issue is Candle does not support excluding pad count in pooling
The average pool config that is being used in the squeezenet is using .with_count_include_pad(false) which is not supported in Candle.
The average pool config that is being used in the squeezenet is using
.with_count_include_pad(false)which is not supported in Candle.
We can fix this by:
- Modify Candle backend with Zero padding the input before average pool (this is what PyTorch did).
- File a ticket with Candle team to enhance the avr pool.
The average pool config that is being used in the squeezenet is using
.with_count_include_pad(false)which is not supported in Candle.We can fix this by:
- Modify Candle backend with Zero padding the input before average pool (this is what PyTorch did).
- File a ticket with Candle team to enhance the avr pool.
@louisfd do you think it's possible?