README.md

# rnn compression
  using mpo to compress rnn


  main.py is old, which only can use gru model


  main_all.py can use gru，my transformer, pytorch transformer on random pattern and tang poem dataset 


  main_pyt.py can use gru, my transformer, pytorch transformer on wikitext .

  
  for example:

  python main_pyt.py   --epoch 200  --model_type mytransf  --lr 5 --optimizer sgd   --N 2  --batch_size 20 --d_model 200 --d_ff 200 --h 2