Skip to content

Sequence Level Training with Deep Reinforcement Learning

Notifications You must be signed in to change notification settings

harmanpreet93/nmt-with-rl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

42 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Reinforcement Learning for Sequence to Sequence Models

Video

Video

Results

alt text

How to train model

Set your configuration parameters in code/user_config.json and to start training run script bash scripts/train_model.sh

TODO:

  • Add beam search
  • Label Smoothing
  • Run back-translation iteration for low resource machine translation scenario

Authors

Akshay Singh Rana - 260963467
Harmanpreet Singh - 260962547