Skip to content

Adapt cnncreator template

Carlos Alfredo Yeverino Rodriguez requested to merge adapt_cnncreator_template into master

Modified CNNCreator.ftl: -Deleted dummy data and label -Added correct function add_input so lmdb data sets can be read as input for the networks. -Save the deployment model instead of the test model to avoid including the lmdb reader. -Adapted functions so they can take the data and label values from the function add_input. -Included training parameters EPOCHS and BATCH_SIZE; they do not apply for testing. -Included a proper image test. -Improved print messages.

Adapted input and output layer templates to fix errors while running caffe2

Corrected tests according to changes in the cnncreator, input and output layer templates

Merge request reports