lamp-data
lamp-data
API
lamp.data
bert
bytesegmentencoding
ByteSegmentCodec
ByteSegmentCodecFactory
distributed
DistributedCommunication
Command
Stop
Train
Valid
DistributedCommunicationNonRoot
DistributedCommunicationRoot
LocalCommunication
LocalIOCommunicationClient
LocalIOCommunicationServer
LoopState
LoopState
LoopStateWithModelAndOptimizerData
LoopStateWithModelAndOptimizerData
languagemodel
schemas
ByteSegmentEncoding
ByteSegmentEncoding
LearningRateScheduleSchemas
LoopState
LoopState
SWALoopState
SimpleLoopState
SimpleThenSWALoopState
TensorDescriptor
TensorDescriptor
TensorList
TensorList
BatchStream
BatchStream
StagedLoader
BucketState
NotYetOpen
Opened
BufferedImageHelper
Codec
CodecFactory
DataParallel
EmptyBatch
EndStream
GraphBatchStream
Graph
IOLoops
TrainingLoopContext
TrainingLoopContext
IdentityCodec
IdentityCodecFactory
LoopState
NonEmptyBatch
Peek
Reader
SWA
SWALearningRateSchedule
SWALearningRateSchedule
SWALoopState
SimpleLoopState
SimpleThenSWALoopState
StateIO
StreamControl
StreamControl
Text
TrainingCallback
TrainingCallback
ValidationCallback
ValidationCallback
Writer
lamp-data
/
lamp.data
/
DataParallel
DataParallel
lamp.data.DataParallel$
object
DataParallel
Attributes
Graph
Reset zoom
Hide graph
Show graph
Supertypes
class
Object
trait
Matchable
class
Any
Self type
DataParallel
.type
Members list
Clear all
Value members
Concrete methods
def
oneEpoch
[
I
,
M
<:
GenericModule
[
I
,
Variable
],
S
,
C
](
epochCount
:
Long
,
trainingCallback
:
TrainingCallback
,
mainModel
:
ModelWithOptimizer
[
I
,
M
],
trainBatches
:
BatchStream
[(
I
,
STen
),
S
,
C
],
logger
:
Option
[
Logger
],
learningRateScheduleFactor
:
Double
,
models
:
Seq
[
SupervisedModel
[
I
,
M
]],
accumulateGradientOverNBatches
:
Int
):
IO
[
Double
]
Updates main model in place, returns average training loss
Updates main model in place, returns average training loss
Attributes
def
validationOneEpoch
[
I
,
M
<:
GenericModule
[
I
,
Variable
],
S
,
C
](
models
:
Seq
[
SupervisedModel
[
I
,
M
]],
validationBatches
:
BatchStream
[(
I
,
STen
),
S
,
C
],
validationCallback
:
ValidationCallback
,
logger
:
Option
[
Logger
],
epochCount
:
Long
):
IO
[
Double
]
In this article
Attributes
Members list
Value members
Concrete methods