Skip to content

SKNahin/Unified-BART

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Unified-BART : Multi-purpose Text Classification and Generation Model

This repository contains a complete setup for training and evaluating a Unified BART model using custom dataset handling and trainer modifications. (The checkpoints and data is not included in the repository yet. It will be released soon.)

Paths

  • data/: Directory to store datasets.
  • models/: Saved model checkpoints and evaluation metrics.
  • src/: Source code files including model, dataset, and evaluation scripts.

Usage

  1. Training the Model To train the model, use the following command:

    python src/train.py
  2. Evaluating the Model To evaluate the model, run:

    python src/evaluate.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages