Merge pull request #1215 from 2017pxy/master

FIX: fix the neg_sampling config
RUCAIBox · Mar 29, 2022 · 430db16 · 430db16
2 parents 4409968 + cab979e
commit 430db16
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:
+        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):