model.json

{
  "architecture": "nampnn",
  "dropout": 0.1,
  "feature_size": 128,
  "hidden_size": 128,
  "include_predicted_na_n": true,
  "num_decoder_blocks": 3,
  "num_encoder_blocks": 3,
  "num_neighbors": 32,
  "num_rbf": 16,
  "positional_embedding_size": 16
}