LujainAbdulrahman commited on
Commit
84fcdb0
·
verified ·
1 Parent(s): d8f0d83

Initial commit of the custom Seq2SeqTransformer model

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1 +1 @@
1
- {"vocab_size": 30522, "batch_size": 16, "d_model": 512, "num_heads": 8, "num_encoder_layers": 6, "num_decoder_layers": 6, "max_len": 64, "lr": 0.0001, "num_epochs": 10}
 
1
+ {"model_type": "custom_seq2seq", "architecture": "custom_seq2seq", "vocab_size": 32000, "batch_size": 16, "d_model": 512, "num_heads": 8, "num_encoder_layers": 6, "num_decoder_layers": 6, "max_len": 64, "lr": 0.0001, "num_epochs": 10}