BertPretrainModule

lamp.nn.bert.BertPretrainModule$
See theBertPretrainModule companion class

Attributes

Companion
class
Graph
Supertypes
trait Product
trait Mirror
class Object
trait Matchable
class Any
Self type

Members list

Type members

Inherited types

type MirroredElemLabels <: Tuple

The names of the product elements

The names of the product elements

Attributes

Inherited from:
Mirror
type MirroredLabel <: String

The name of the type

The name of the type

Attributes

Inherited from:
Mirror

Value members

Concrete methods

def apply[S : Sc](maxLength: Int, vocabularySize: Int, segmentVocabularySize: Int, mlmHiddenDim: Int, wholeStentenceHiddenDim: Int, numBlocks: Int, embeddingDim: Int, attentionHiddenPerHeadDim: Int, attentionNumHeads: Int, bertEncoderMlpHiddenDim: Int, dropout: Double, tOpt: STenOptions, linearized: Boolean, positionEmbedding: Option[STen]): BertPretrainModule

Implicits

Implicits

implicit val load: Load[BertPretrainModule]