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