examples/dcgan/dcgan.rb in ruby-dnn-0.15.3 vs examples/dcgan/dcgan.rb in ruby-dnn-0.16.0

- old
+ new

@@ -17,11 +17,11 @@ @bn4 = BatchNormalization.new @bn5 = BatchNormalization.new @bn6 = BatchNormalization.new end - def call(x) + def forward(x) x = InputLayer.new(20).(x) x = @l1.(x) x = @bn1.(x) x = ReLU.(x) @@ -61,11 +61,11 @@ @l4 = Conv2D.new(64, 4, padding: true) @l5 = Dense.new(1024) @l6 = Dense.new(1) end - def call(x, trainable = true) + def forward(x, trainable = true) @l1.trainable = trainable @l2.trainable = trainable @l3.trainable = trainable @l4.trainable = trainable @l5.trainable = trainable @@ -101,10 +101,10 @@ super() @gen = gen @dis = dis end - def call(x) + def forward(x) x = @gen.(x) x = @dis.(x, false) x end