Add comment for an inputColorMode field (#400)

Kotlin · Jul 12, 2022 · e89f88a · e89f88a
1 parent fc8f349
commit e89f88a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModels.kt b/api/src/main/kotlin/org/jetbrains/kotlinx/dl/api/inference/keras/loaders/TFModels.kt
@@ -574,6 +574,10 @@ public interface ModelType<T : InferenceModel, U : InferenceModel> {
      */
     public val channelsFirst: Boolean
 
+    /**
+      * An expected channels order for the input image.
+      * Note: the wrong choice of this parameter can significantly impact the model's performance.
+     */
     public val inputColorMode: ColorMode
 
     /**