Skip to content

Instantly share code, notes, and snippets.

@blythed
Created April 6, 2022 12:05
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save blythed/d45c84b4c6128fdf5ac1cd2823211f52 to your computer and use it in GitHub Desktop.
Save blythed/d45c84b4c6128fdf5ac1cd2823211f52 to your computer and use it in GitHub Desktop.
import models
models.build_classifier.rnn_layer = @LSTM()
models.build_classifier.n_tokens = 16
LSTM.hidden_size = 512
LSTM.input_size = 64
LSTM.num_layers = 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment