update codes

bigdata-ustc · Sep 25, 2024 · 4c39b42 · 4c39b42
1 parent 408b5dd
commit 4c39b42
Show file tree

Hide file tree

Showing 3 changed files with 98 additions and 2 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/MIRT/train.py b/MIRT/train.py
@@ -55,8 +55,8 @@ def train():
                 running_loss = 0.0
 
         # validate and save current model every epoch
-        rmse, auc = validate(net, epoch)
         save_snapshot(net, './model/' + data + '/model_epoch' + str(epoch + 1))
+        validate(net, epoch)
 
 
 def validate(model, epoch):

diff --git a/TechCD/train.py b/TechCD/train.py
@@ -69,7 +69,7 @@ def train():
 
         # validate and save current model every epoch
         save_snapshot(net, './model/' + data + '/model_epoch' + str(epoch + 1))
-        rmse, auc = validate(net, epoch)
+        validate(net, epoch)
 
 
 def validate(net, epoch):