Packages

class ONNXHub extends Logging

Linear Supertypes
Logging, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ONNXHub
  2. Logging
  3. AnyRef
  4. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ONNXHub(modelCacheDir: Path)
  2. new ONNXHub(connectTimeout: Int = ONNXHub.DefaultConnectTimeout, readTimeout: Int = ONNXHub.DefaultReadTimeout, retryCount: Int = ONNXHub.DefaultRetryCount, retryTimeoutInSeconds: Int = ...)
  3. new ONNXHub(modelCacheDir: Path, connectTimeout: Int, readTimeout: Int, retryCount: Int, retryTimeoutInSeconds: Int)

Value Members

  1. val connectTimeout: Int
  2. def getDir: Path
  3. def getModelInfo(model: String, repo: String = ONNXHub.DefaultRepo, opset: Option[Int] = None): ONNXModelInfo
  4. def listModels(repo: String = ONNXHub.DefaultRepo, model: Option[String] = None, tags: Option[Seq[String]] = None): Seq[ONNXModelInfo]
  5. def load(model: String, repo: String = ONNXHub.DefaultRepo, opset: Option[Int] = None, forceReload: Boolean = false, silent: Boolean = false, allowUnsafe: Boolean = false): Array[Byte]
  6. val modelCacheDir: Path
  7. val readTimeout: Int
  8. val retryCount: Int
  9. val retryTimeoutInSeconds: Int