Advanced Computing Platform for Theoretical Physics

Commit 0f8ed42e authored by Pengfei Zhou's avatar Pengfei Zhou
Browse files

Update README.md

parent b325dde9
# 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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment