Types & classes — github.com/IBM/PoWER-BERT

CodeHub MCPcopy Create free account

hub / github.com/IBM/PoWER-BERT / types & classes

Types & classes10 in github.com/IBM/PoWER-BERT

⨍ Functions 84 ◇ Types & classes 10

↓ 8 callersClassAdamWarmup

Adam optimizer with warmup. Default parameters follow those provided in the original paper. # Arguments decay_steps: Learning rate wil

utils/Adam_mult.py:31

↓ 1 callersClassCustomConstraint

model/model.py:34

↓ 1 callersClassHard_Extract

model/model.py:111

↓ 1 callersClassMultiHeadAttention

model/multi_head_attention.py:13

↓ 1 callersClassScaledDotProductAttention

model/scaled_dot_attention.py:16

↓ 1 callersClassSoft_Extract

model/model.py:47

↓ 1 callersClassdata_parser

utils/data_parser.py:11

↓ 1 callersClasstraining

ClassConstraint

model/model.py:26

ClassTest_PowerBERTModels

test/test_model.py:15