Unable to CPP Export LeNet

CEA-LIST / N2D2

N2D2 is an open source CAD framework for Deep Neural Network simulation and full DNN-based applications building.

Other

147 stars 35 forks source link

Unable to CPP Export LeNet #82

Closed stephaneburel-cea closed 3 years ago

stephaneburel-cea commented 3 years ago

Hello. I come to report a bug for CPP export. When exporting the simple Lenet network : .n2d2.sh ../../models/LeNet.ini -export CPP -calib -1 -seed 1 The export fails. The log is : error.txt

Best Regards, Stéphane Burel

olivierbichler-cea commented 3 years ago

Hi, The export fail because the CPP export currently doesn't support the softmax layer. Indeed, the softmax is hard to quantize and is not useful for inference. The solution is to remove this layer before export. Olivier

olivierbichler-cea commented 3 years ago

Closing the issue as the tool already generate a proper error message.