class DNN::Savers::JSONSaver

Private Instance Methods

dump_bin() click to toggle source
# File lib/dnn/core/savers.rb, line 115
def dump_bin
  data = { version: VERSION, class: @model.class.name, params: get_all_params_base64_data }
  JSON.dump(data)
end
get_all_params_base64_data() click to toggle source
# File lib/dnn/core/savers.rb, line 120
def get_all_params_base64_data
  @model.trainable_layers.map do |layer|
    layer.get_params.to_h do |key, param|
      base64_data = Base64.encode64(param.data.to_binary)
      [key, [param.data.shape, base64_data]]
    end
  end
end