hub / github.com/IBM/PoWER-BERT

github.com/IBM/PoWER-BERT @main

Chat with this repo

repository ↗ · DeepWiki ↗ · + Follow

94 symbols 307 edges 13 files 24 documented · 26% ● updated 11mo ago★ 632 open issues

Browse by type

⨍ Functions 84 ◇ Types & classes 10

What it actually does AI analysis from the code graph — generated when you open this

loading…

Core symbols most depended-on inside this repo

↓ 4 filesAdamWarmup

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

utils/Adam_mult.py

↓ 3 filesbuild_model_from_config

Build the model from config file. :param config_file: The path to the JSON configuration file. :param training: If training, the …

model/checkpoint_loader.py

↓ 2 filesretention_config_parser

Parse the retention configuration provided by the user. Also, check the provided rentention config for correctness.

utils/retention_parser.py

↓ 2 filescalc_train_steps

Calculate the number of total and warmup steps. >>> calc_train_steps(num_example=1024, batch_size=32, epochs=10, …

utils/Adam_mult.py

↓ 2 filesload_checkpoint

Load trained official model from checkpoint. :param model: Built keras model. :param config: Loaded configuration file. :param …

model/checkpoint_loader.py

browse all functions →

Shape

Method 60

Function 24

Class 10

Languages

Python100%

Modules by API surface

model/model.py30 symbols

utils/data_parser.py17 symbols

model/multi_head_attention.py10 symbols

model/checkpoint_loader.py9 symbols

utils/Adam_mult.py7 symbols

training.py6 symbols

model/scaled_dot_attention.py6 symbols

test/test_model.py4 symbols

prediction.py2 symbols

utils/retention_parser.py1 symbols

utils/mean_squared_error.py1 symbols

evaluation.py1 symbols

For agents

$ claude mcp add PoWER-BERT \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact

github.com/IBM/PoWER-BERT @main

Core symbols most depended-on inside this repo

Shape

Languages

Modules by API surface

For agents

Ask about this repo answers extend the page