case class DataLoaderConfig(batchSize: Int = 16, prefetchBatches: Int = 20, shuffleInPartition: Boolean = true, timeoutMillis: Long = 10000) extends Product with Serializable
Configuration for the NerDLDataLoader.
- batchSize
Number of sentences per batch (default: 16)
- prefetchBatches
Number of batches to prefetch per worker (default: 2). Total prefetch buffer size will be numWorkers * prefetchFactor
- shuffleInPartition
Whether to shuffle the data (default: true). Improves training convergence.
- timeoutMillis
Timeout in milliseconds for fetching a batch (default: 10000). Prevents hanging on slow operations.
Linear Supertypes
Ordering
- Alphabetic
- By Inheritance
Inherited
- DataLoaderConfig
- Serializable
- Serializable
- Product
- Equals
- AnyRef
- Any
- Hide All
- Show All
Visibility
- Public
- All
Instance Constructors
-
new
DataLoaderConfig(batchSize: Int = 16, prefetchBatches: Int = 20, shuffleInPartition: Boolean = true, timeoutMillis: Long = 10000)
- batchSize
Number of sentences per batch (default: 16)
- prefetchBatches
Number of batches to prefetch per worker (default: 2). Total prefetch buffer size will be numWorkers * prefetchFactor
- shuffleInPartition
Whether to shuffle the data (default: true). Improves training convergence.
- timeoutMillis
Timeout in milliseconds for fetching a batch (default: 10000). Prevents hanging on slow operations.
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
- val batchSize: Int
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
- val prefetchBatches: Int
- val shuffleInPartition: Boolean
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
- val timeoutMillis: Long
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()