class DNN::Savers::MarshalSaver

Public Class Methods

new(model, include_model: true) click to toggle source
Calls superclass method DNN::Savers::Saver::new
# File lib/dnn/core/savers.rb, line 90
def initialize(model, include_model: true)
  super(model)
  @include_model = include_model
end

Private Instance Methods

dump_bin() click to toggle source
# File lib/dnn/core/savers.rb, line 95
        def dump_bin
  params_data = @model.get_all_params_data
  if @include_model
    @model.clean_layers
    data = {
      version: VERSION, class: @model.class.name,
      params: params_data, model: @model
    }
  else
    data = { version: VERSION, class: @model.class.name, params: params_data }
  end
  bin = Zlib::Deflate.deflate(Marshal.dump(data))
  @model.set_all_params_data(params_data) if @include_model
  bin
end