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