From cab979e869778d4f5902475d52e261e9d83cc0aa Mon Sep 17 00:00:00 2001 From: 2017pxy <2017202006@ruc.edu.cn> Date: Tue, 29 Mar 2022 11:15:43 +0800 Subject: [PATCH] FIX: fix the bug in neg_sampling config --- recbole/config/configurator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recbole/config/configurator.py b/recbole/config/configurator.py index cd1733ae6..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 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):