CoolFace
Modelpublic

Pradheep1647/meeting_summarization_kda-meetingbank-bs8-e20-bf16-4

sourceHugging Facemitupdated 2mo agoView on Hugging Face
0likes4downloads
Model Card

Meeting Summarization Kda

Custom PyTorch Transformer checkpoint trained on MeetingBank for meeting summarization research. This repository is part of the `transformer-lab` collection.

Model Details

FieldValue
RepositoryPradheep1647/meeting_summarization_kda-meetingbank-bs8-e20-bf16-4
Attentionkda
Datasetmeetingbank
Layers6
Hidden size512
Heads8
Batch size1
Effective batch size8
Epochs20
Precisionbf16
Checkpointmeeting_model_kda04.pt
Optimizer steps12,920
Logged training time59m 46s

Architecture

[image]

Static architecture diagram generated from this run's config.json, including model width, depth, sequence dimensions, and attention-specific settings.

Training Loss

[image]

Raw curve data is available in `loss_curve.csv`.

The curve covers the complete training run. The uploaded checkpoint is the saved epoch with the lowest full-validation loss, not simply the last epoch.

Evaluation

MetricValue
Validation loss2.5381
Perplexity12.6559
Token accuracy0.5497
Top-5 accuracy0.7400
ROUGE-10.2556
ROUGE-20.0853
ROUGE-L0.2055
BLEU7.90
Evaluation tokens/s4789.1
Generation tokens/s92.9
Forward latency (ms)13.17
Peak CUDA memory (MB)189.5

Core metrics use the full MeetingBank validation split. Generation metrics use the first 128 validation examples with greedy decoding.

Available Models

Files

FilePurpose
meeting_model_kda04.ptPyTorch checkpoint containing model_state_dict, optimizer states, epoch, and global step.
config.jsonTraining and architecture config converted from the Hydra run config.
architecture.pngArchitecture diagram generated from the saved model config, with block shapes and dimensions.
tokenizer.jsonUnified MeetingBank transcript and summary tokenizer.
causal_tokenizer.jsonExplicit alias of the unified causal tokenizer.
loss_curve.csvTensorBoard train/loss scalar export.
loss_curve.svgStatic training-loss plot generated from loss_curve.csv.

Usage

These checkpoints are from a custom PyTorch codebase, not a transformers.AutoModel checkpoint. Use the repo-native builder to instantiate the architecture, then load the checkpoint state dict.

python
from pathlib import Path

import torch
from huggingface_hub import hf_hub_download
from omegaconf import OmegaConf

import src  # registers components
from src.model.builder import build_causal_lm

repo_id = "Pradheep1647/meeting_summarization_kda-meetingbank-bs8-e20-bf16-4"

config_path = hf_hub_download(repo_id=repo_id, filename="config.json")
checkpoint_path = hf_hub_download(repo_id=repo_id, filename="meeting_model_kda04.pt")

cfg = OmegaConf.load(config_path)
model = build_causal_lm(cfg)

state = torch.load(checkpoint_path, map_location="cpu")
model.load_state_dict(state["model_state_dict"])
model.eval()

print(f"Loaded {repo_id} from {Path(checkpoint_path).name}")

Notes

  • —This is a research checkpoint for comparing attention variants under the same MeetingBank setup.
  • —The config and tokenizers are included so future runs can reproduce the architecture and preprocessing assumptions.
  • —Use config.json as the source of truth for architecture parameters.