modified project

linjing-lab · Oct 29, 2023 · ce33de2 · ce33de2
1 parent 5db7c84
commit ce33de2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/released_box/perming/common.py b/released_box/perming/common.py
@@ -174,7 +174,7 @@ def __init__(self,
                                        batch_size, 
                                        learning_rate_init, 
                                        lr_scheduler)
-        assert num_classes >= 2, 'The predefined options of Multipler are more suitable for Multi-classification.'
+        assert num_classes >= 2
 
     def _activate(self, activation: str):
         '''

diff --git a/released_box/perming/general.py b/released_box/perming/general.py
@@ -91,13 +91,13 @@ def _criterion(self, criterion: str):
             return torch.nn.CrossEntropyLoss()
         elif criterion == 'NLLLoss':
             return torch.nn.NLLLoss()
-        elif criterion == 'MultiLabelSoftMarginLoss':
+        elif criterion == 'MultiLabelSoftMarginLoss': # multi-outputs
             return torch.nn.MultiLabelSoftMarginLoss()
         elif criterion == 'BCELoss': # classification with num_classes = 2
             return torch.nn.BCELoss()
-        elif criterion == 'BCEWithLogitsLoss':
+        elif criterion == 'BCEWithLogitsLoss': # multi-outputs
             return torch.nn.BCEWithLogitsLoss()
-        elif criterion == 'MSELoss': # regression
+        elif criterion == 'MSELoss': # regression (or multi-outputs)
             return torch.nn.MSELoss()
         elif criterion == 'L1Loss':
             return torch.nn.L1Loss()