Commit 80325d10 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Classification head of dapper models consistent with adapter dimension

parent 1dd689cc
......@@ -24,12 +24,12 @@ class DAPPER_model(nn.Module):
)
self.fc_moba = nn.Sequential(
nn.BatchNorm1d(1000),
nn.Linear(1000, 1000),
nn.BatchNorm1d(adapter_out_features),
nn.Linear(adapter_out_features, adapter_out_features),
nn.ReLU(True),
nn.Dropout(dropout),
nn.BatchNorm1d(1000),
nn.Linear(1000, 256),
nn.BatchNorm1d(adapter_out_features),
nn.Linear(adapter_out_features, 256),
nn.ReLU(True),
nn.Dropout(dropout),
)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment