nn_benchmark
nn_benchmark copied to clipboard
quantVGG16 model for input size(1,3,224,224)
Hi, I try to train the model quantvgg16 with the dataset ILSVRC2012 , but can not get a Convergent model. did you have a try ? for the dataset ILSVRC2012, the classifier is as below:
self.classifier = nn.Sequential(
make_quant_linear(512 * 7 * 7, 4096, bias=True, bit_width=bit_width),
make_quant_relu(bit_width),
nn.Dropout(),
make_quant_linear(4096, 4096, bias=True, bit_width=bit_width),
make_quant_relu(bit_width),
nn.Dropout(),
make_quant_linear(4096, num_classes, bias=False, bit_width=bit_width,
weight_scaling_per_output_channel=False),
)
I also tried with your code for mnist, it will be quickly up to 100% acc!
thanks!
Hi, I will put an announcement in the readme but I don't plan on maintaining this repo I am sorry... I wrote a simple wrapper around Brevitas to conduct simple experiments with the base MLP examples FINN provided but do not mean to replace the examples themselves (https://github.com/Xilinx/finn-examples). I would advise you to ask your questions on the gitter channels or in the FINN/Brevitas repositories directly! As for your case in particular, I remember I could not make VGG/Mobilenet to work but planned on trying to port them later on (which eventually never happened). In conclusion, sorry but I cannot help you! ☹️