diff --git a/recbole/config/configurator.py b/recbole/config/configurator.py index e61853778..7f46b6c09 100644 --- 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):