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