combing training script

Author: dmitryduev

scripts/train.py

@@ -204,7 +204,7 @@ def train_and_eval(
     classifier = tails.models.DNN(name=model_name)
 
     tails_loss = TailsLoss(name="loss", w_1=w_1, w_2=w_2)
-    label_accuracy = LabelAccuracy(threshold=0.5)
+    label_accuracy = LabelAccuracy(threshold=class_threshold)
     # convert position RMSE to pixels
     position_rmse = PositionRootMeanSquarredError(scaling_factor=scaling_factor)
 
@@ -243,7 +243,8 @@ def train_and_eval(
     if verbose:
         log(stats)
 
-    # classifier.model.save_weights('tails')
+    if save_model:
+        classifier.model.save_weights(f"{model_name}-{tag}")
 
 
 if __name__ == "__main__":