nn.Linear(in_features = dim_model, out_features = dim_hidden),
nn.ReLU(),
nn.Linear(in_features = dim_hidden, out_features = dim_model),
nn.Linear(in_features = dim_model, out_features = dim_hidden),
nn.ReLU(),
nn.Linear(in_features = dim_hidden, out_features = dim_model),