Skip to content
Snippets Groups Projects
Unverified Commit 12d6b7e2 authored by Lev Zlotnik's avatar Lev Zlotnik Committed by GitHub
Browse files

Added convert_model_to_distiller_lstm (#259) 

* Traverses recursively through entire model and replaces all submodules of type `nn.LSTM` and `nn.LSTMCell` with distiller versions
parent a3f2ce2d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
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