FIX: fix the bug in neg_sampling config

RUCAIBox · Mar 29, 2022 · cab979e · cab979e
1 parent 7eabb09
commit cab979e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recbole/config/configurator.py b/recbole/config/configurator.py
@@ -349,7 +349,7 @@ def _init_device(self):
 
     def _set_train_neg_sample_args(self):
         neg_sampling = self.final_config_dict['neg_sampling']
-        if neg_sampling is None or neg_sampling.lower() == 'none':
+        if neg_sampling is None or neg_sampling == 'None':
             self.final_config_dict['train_neg_sample_args'] = {'strategy': 'none'}
         else:
             if not isinstance(neg_sampling, dict):