input_dim: 768
loss: cosine_embedding_loss
num_units:
- 512
- 256
- 128
optimizer: Adam
output_dim: 100