(self, model_dim, max_len, num_layer, num_head, n_vocab, lr, max_seg=3, drop_rate=0.2,padding_idx=0)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected