features.embeddings.linear
features.edge_embedding.weight
features.norm_edges
W_e
W_s.weight
encoder_layers.0.norm1
encoder_layers.0.norm2
encoder_layers.0.norm3
encoder_layers.0.W1
encoder_layers.0.W2
encoder_layers.0.W3
encoder_layers.0.W11
encoder_layers.0.W12
encoder_layers.0.W13
encoder_layers.0.dense.W_in
encoder_layers.0.dense.W_out
encoder_layers.1.norm1
encoder_layers.1.norm2
encoder_layers.1.norm3
encoder_layers.1.W1
encoder_layers.1.W2
encoder_layers.1.W3
encoder_layers.1.W11
encoder_layers.1.W12
encoder_layers.1.W13
encoder_layers.1.dense.W_in
encoder_layers.1.dense.W_out
encoder_layers.2.norm1
encoder_layers.2.norm2
encoder_layers.2.norm3
encoder_layers.2.W1
encoder_layers.2.W2
encoder_layers.2.W3
encoder_layers.2.W11
encoder_layers.2.W12
encoder_layers.2.W13
encoder_layers.2.dense.W_in
encoder_layers.2.dense.W_out
decoder_layers.0.norm1
decoder_layers.0.norm2
decoder_layers.0.W1
decoder_layers.0.W2
decoder_layers.0.W3
decoder_layers.0.dense.W_in
decoder_layers.0.dense.W_out
decoder_layers.1.norm1
decoder_layers.1.norm2
decoder_layers.1.W1
decoder_layers.1.W2
decoder_layers.1.W3
decoder_layers.1.dense.W_in
decoder_layers.1.dense.W_out
decoder_layers.2.norm1
decoder_layers.2.norm2
decoder_layers.2.W1
decoder_layers.2.W2
decoder_layers.2.W3
decoder_layers.2.dense.W_in
decoder_layers.2.dense.W_out
W_out