add_model_train

LearnMF · Apr 28, 2021 · 05cd475 · 05cd475
1 parent e6795be
commit 05cd475
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/model/model/Torch_model/Souhu_TextMatch/Conditional_Norm.py b/model/model/Torch_model/Souhu_TextMatch/Conditional_Norm.py
@@ -134,6 +134,7 @@ def forward(self, input_ids, attention_mask,token_type_ids,c):
 def train(dataloader, model, loss_fn, optimizer):
     size = len(dataloader.dataset)
     correct = 0
+    model.train()
     for batch, (data,y,c) in enumerate(dataloader):
         input_ids = data['input_ids'].to(device)
         attention_mask = data['attention_mask'].to(device)

diff --git a/model/model/Torch_model/Souhu_TextMatch/Nezha_1024.py b/model/model/Torch_model/Souhu_TextMatch/Nezha_1024.py
@@ -144,6 +144,7 @@ def forward(self, input_ids, attention_mask, token_type_ids, c):
 def train(dataloader, model, loss_fn, optimizer):
     size = len(dataloader.dataset)
     correct = 0
+    model.train()
     for batch, (data, y, c) in enumerate(dataloader):
         input_ids = data['input_ids'].to(device)
         attention_mask = data['attention_mask'].to(device)

diff --git a/model/model/Torch_model/Souhu_TextMatch/P-tuning.py b/model/model/Torch_model/Souhu_TextMatch/P-tuning.py
@@ -123,6 +123,7 @@ def forward(self, input_ids, attention_mask, token_type_ids):
 def train(dataloader, model, loss_fn, optimizer):
     size = len(dataloader.dataset)
     correct = 0
+    model.train()
     for batch, (data,y,_) in enumerate(dataloader):
         input_ids = data['input_ids'].to(device)
         attention_mask = data['attention_mask'].to(device)