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