提交 94ca8a36 作者: 朱学凯

change mlp

上级 d5064e86
...@@ -2,16 +2,7 @@ ...@@ -2,16 +2,7 @@
<project version="4"> <project version="4">
<component name="ChangeListManager"> <component name="ChangeListManager">
<list default="true" id="d29948e3-1642-45ab-9fe2-087a876b83b3" name="Default Changelist" comment=""> <list default="true" id="d29948e3-1642-45ab-9fe2-087a876b83b3" name="Default Changelist" comment="">
<change afterPath="$PROJECT_DIR$/.idea/inspectionProfiles/Project_Default.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/CPI.iml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/CPI.iml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/misc.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/misc.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/modules.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/modules.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/shelf/Uncommitted_changes_before_Update_at_2021_4_11,_11_03_上午_[Default_Changelist]/shelved.patch" beforeDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/shelf/Uncommitted_changes_before_Update_at_2021_4_11__11_03___Default_Changelist_.xml" beforeDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/vcs.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/vcs.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" /> <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/dataset.py" beforeDir="false" afterPath="$PROJECT_DIR$/dataset.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/modeling_bert.py" beforeDir="false" afterPath="$PROJECT_DIR$/modeling_bert.py" afterDir="false" />
<change beforePath="$PROJECT_DIR$/run_interaction.py" beforeDir="false" afterPath="$PROJECT_DIR$/run_interaction.py" afterDir="false" /> <change beforePath="$PROJECT_DIR$/run_interaction.py" beforeDir="false" afterPath="$PROJECT_DIR$/run_interaction.py" afterDir="false" />
</list> </list>
<option name="SHOW_DIALOG" value="false" /> <option name="SHOW_DIALOG" value="false" />
...@@ -97,7 +88,7 @@ ...@@ -97,7 +88,7 @@
<option name="number" value="Default" /> <option name="number" value="Default" />
<option name="presentableId" value="Default" /> <option name="presentableId" value="Default" />
<updated>1618489611823</updated> <updated>1618489611823</updated>
<workItem from="1618489616148" duration="6353000" /> <workItem from="1618489616148" duration="6549000" />
</task> </task>
<servers /> <servers />
</component> </component>
......
...@@ -1827,20 +1827,20 @@ class BertForQuestionAnswering(BertPreTrainedModel): ...@@ -1827,20 +1827,20 @@ class BertForQuestionAnswering(BertPreTrainedModel):
class Multilayer_perceptron(nn.Module): class Multilayer_perceptron(nn.Module):
def __init__(self, config): def __init__(self, config):
super(Multilayer_perceptron, self).__init__() super(Multilayer_perceptron, self).__init__()
self.layer_1 = nn.Linear(config.hidden_size, 256) self.layer_1 = nn.Linear(config.hidden_size, 1)
self.layer_2 = nn.Linear(512, 256) # self.layer_2 = nn.Linear(512, 256)
self.layer_3 = nn.Linear(256, 1) # self.layer_3 = nn.Linear(256, 1)
self.drop_out = nn.Dropout(0.5) # self.drop_out = nn.Dropout(0.5)
def forward(self, bert_output): def forward(self, bert_output):
x1 = self.layer_1(bert_output) x1 = self.layer_1(bert_output)
x1 = F.relu(x1, inplace=True) # x1 = F.relu(x1, inplace=True)
x1 = self.drop_out(x1) # x1 = self.drop_out(x1)
# x2 = self.layer_2(x1) # x2 = self.layer_2(x1)
# x2 = F.relu(x2, inplace=True) # x2 = F.relu(x2, inplace=True)
# x2 = self.drop_out(x2) # x2 = self.drop_out(x2)
x2 = self.layer_3(x1) # x2 = self.layer_3(x1)
return x2 return x1
class BertAffinityModel(BertPreTrainedModel): class BertAffinityModel(BertPreTrainedModel):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论