Minor fixes

SoftServeInc · Jun 14, 2022 · 8085ecf · 8085ecf
1 parent 82f0994
commit 8085ecf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/chemprop/utils.py b/chemprop/utils.py
@@ -167,7 +167,7 @@ def get_loss_func(args: Namespace) -> nn.Module:
         return nn.BCEWithLogitsLoss(reduction='none')
 
     if args.dataset_type == 'regression':
-        return nn.L1Loss(reduction='none')
+        return nn.L2Loss(reduction='none')
 
     if args.dataset_type == 'multiclass':
         return nn.CrossEntropyLoss(reduction='none')

diff --git a/train.py b/train.py
@@ -1,4 +1,5 @@
-"""Trains a model on a dataset."""
+"""Entrypoint fot training."""
+"""Single or cross-validation training"""
 
 from chemprop.parsing import parse_train_args
 from chemprop.train import cross_validate