lib/dnn/core/model.rb in ruby-dnn-0.6.0 vs lib/dnn/core/model.rb in ruby-dnn-0.6.1
- old
+ new
@@ -3,10 +3,9 @@
module DNN
# This class deals with the model of the network.
class Model
attr_accessor :layers
attr_reader :optimizer
- attr_reader :batch_size
def self.load(file_name)
Marshal.load(File.binread(file_name))
end