comment

EmergentOrder · Oct 14, 2020 · bc00ea0 · bc00ea0
1 parent d2c82b3
commit bc00ea0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/core/src/main/scala/ONNXBytesDataSource.scala b/core/src/main/scala/ONNXBytesDataSource.scala
@@ -8,6 +8,7 @@ class ONNXBytesDataSource(onnxBytes: Array[Byte]) extends AutoCloseable with Dat
 
   val onnxHelper = new ONNXHelper(onnxBytes)
 
+  //TODO: return non-tensor params
   override def getParams[T: Numeric: ClassTag](name: String): Tensor[T] = {
     val params = onnxHelper.params.filter(x => x._1 == name).headOption
     params match {