Skip to content

A flexible pytorch template for Natural Language Processing based on Bert.

Notifications You must be signed in to change notification settings

codeaudit/pytorch-bert-template

 
 

Repository files navigation

pytorch-bert-template

A flexible pytorch template for Natural Language Processing based on Bert.

Now, it just support the NLC (Natural Language Classification), NLI (Natural Language Inference) and other simple classification mission. It will support the NER and Machine Comprehension in the future

Requirements

How to use

Features

Folder Structure

Usage

Config file format

Using config files

Resuming from checkpoints

Using Multiple GPU

Customization

Custom CLI options

Data Loader

Processor

Agent

Model

Testing

Eval

Predict

Additional logging

Validation data

Checkpoints

TensorboardX Visualization

TODOs

  • Rename trainer to agent
  • Finish the document
  • Finish the ParameterSearch Function.
  • Monitor and control the agent by Wechat
  • Realize the NER
  • Realize the Machine Comprehension

Acknowledgments

This project is inspired by the project pytorch-template by victoresque

About

A flexible pytorch template for Natural Language Processing based on Bert.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%