Commit 4147e4d8 authored by Federico Rossi's avatar Federico Rossi

...

parent 36b6f4ad
......@@ -96,6 +96,7 @@ function(tinydnn_print_configuration_summary)
tinydnn_status(" SSE : " USE_SSE AND COMPILER_HAS_SSE_FLAG THEN "Yes" ELSE "No")
tinydnn_status(" AVX : " USE_AVX AND COMPILER_HAS_AVX_FLAG THEN "Yes" ELSE "No")
tinydnn_status(" AVX2 : " USE_AVX2 AND COMPILER_HAS_AVX2_FLAG THEN "Yes" ELSE "No")
tinydnn_status(" SVE : " USE_SVE THEN "Yes" ELSE "No")
tinydnn_status(" Pthread : " USE_PTHREAD THEN "Yes" ELSE "No")
tinydnn_status(" TBB : " USE_TBB AND TBB_FOUND THEN "Yes (ver. ${TBB_INTERFACE_VERSION})" ELSE "No")
tinydnn_status(" OMP : " USE_OMP AND OMP_FOUND THEN "Yes" ELSE "No")
......
......@@ -31,6 +31,10 @@ public:
void save() {
network.save(modelName+".tinymodel");
}
void dumpWeights() {
network.save(modelName+".tinymodel",content_type::weights);
}
tiny_dnn::vec_t infer(tiny_dnn::vec_t in) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment