Added "Model name: $name" line (if name is not null) to summary met…

…hod Kotlin#120
dosier · Jun 15, 2021 · fba4e3f · dosier · Jun 16, 2021 · dosier
1 parent 6347d65
commit fba4e3f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/Functional.kt b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/Functional.kt
@@ -330,7 +330,9 @@ public class Functional(vararg layers: Layer) : GraphTrainableModel(*layers) {
         check(isModelCompiled) { "The model is not compiled yet. Compile the model to use this method." }
 
         logger.info("==========================================================================================================")
-        logger.info("Model: Functional")
+        logger.info("Model type: Functional")
+        if (name != null)
+            logger.info("Model name: $name")
         logger.info("__________________________________________________________________________________________________________")
         logger.info("Layer (type)                           Output Shape              Param #       Connected to               ")
         logger.info("==========================================================================================================")

diff --git a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/Sequential.kt b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/core/Sequential.kt
@@ -167,6 +167,8 @@ public class Sequential(vararg layers: Layer) : GraphTrainableModel(*layers) {
 
         logger.info("===========================================================================")
         logger.info("Model: Sequential")
+        if (name != null)
+            logger.info("Model name: $name")
         logger.info("___________________________________________________________________________")
         logger.info("Layer (type)                           Output Shape              Param #   ")
         logger.info("===========================================================================")