ruby-dnn 1.1.2 → 1.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 41b7dde05615d65bbc09fec893f67c0874a44aaff4dff8243c9ccdf97f5c9d7a
-  data.tar.gz: fea5cb29b787d4cbbba18f34ab2f3f4e6275f8018ebda6c335c9f3564194b109
+  metadata.gz: 91a60f8ec7da64fb5cb69e872734e917ff5ffbb35574dd4a640b1e6739d3f505
+  data.tar.gz: d4b4e6c127c2f41aaa21bf30ba87a0c44989e94f2aec5a32f92e5d99f7947118
 SHA512:
-  metadata.gz: c18e653840184c9590966b18f36a98a843cf2d1d6ced1159c071ca21839476e4d6f0b6b36bb5319414a50c430cef8a1dd66bedc0ab87d16602896dff6006bc2e
-  data.tar.gz: ec7844e9a5bae664e8a5dd88d5684731c2a2bd3149c8131b8cbe606a9e10c30a68f7f8c4904644f931d2616cdc19dc345254934b792df4eb94117ab31c9b06f2
+  metadata.gz: 90800594f4937415ba25fa13678a9e788d187c6ae3d7eb21f9b8ab4e091d491a0bce4117a06e018378c354bf69a651974148015a70271e944ea992530128f366
+  data.tar.gz: 9eb5f4a99cf493784ff13a1a9e74ef6a6ef693e20c3a9d53e4c24780855225ec9383167c36897b5c03afad5b122b35df9642f4ce71bb61901ecea6311706d559

data/lib/dnn/core/models.rb

@@ -522,7 +522,13 @@ module DNN
       # Clean all layers.
       def clean_layers
         layers.each(&:clean)
-        @loss_func.clean
+        if @loss_func.is_a?(Array)
+          @loss_func.each do |lf|
+            lf.clean
+          end
+        else
+          @loss_func.clean
+        end
         @layers_cache = nil
       end
 

data/lib/dnn/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "1.1.2"
+  VERSION = "1.1.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 1.1.2
+  version: 1.1.3
 platform: ruby
 authors:
 - unagiootoro