If you'd like to run the DOC + Partial Shuffle models, use the same commands as in the original DOC model, presented here.
The code for the Partial Shuffle method itself is in
partial_shuffle.py. If you'd like to use this method in your own language model, simply import
partial_shuffle.py, and call it before each epoch, as in line 196 in
main.py. No other modifications are required.