ModelIO.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>
#include <string>
 
#include "FlexNN.h"
 
namespace FlexNN {
 
struct Status {
  bool ok = true;
  std::string error;
 
  static Status Ok() { return {true, ""}; }
  static Status Err(std::string msg) { return {false, std::move(msg)}; }
};
 
struct ExportOptions {
  int formatVersion = 2; 
};
 
[[nodiscard]] Status exportModel(const NeuralNetwork& net, const std::string& path,
                                 ExportOptions opts = {});
 
[[nodiscard]] Status importModel(NeuralNetwork& net, const std::string& path);
 
} // namespace FlexNN