diff --git a/core/trainers/single_trainer.py b/core/trainers/single_trainer.py index 56741af5ebaa2bd40b94eba625b6449ac42e489b..274047a0127d6ef2df601284cb9cd1e67335cf85 100755 --- a/core/trainers/single_trainer.py +++ b/core/trainers/single_trainer.py @@ -68,7 +68,7 @@ class SingleTrainer(TranspileTrainer): reader = os.path.join(abs_dir, '../utils', 'dataset_instance.py') sparse_slots = envs.get_global_env(name + "sparse_slots", "").strip() dense_slots = envs.get_global_env(name + "dense_slots", "").strip() - if sparse_slots != "" and dense_slots != "": + if sparse_slots == "" and dense_slots == "": pipe_cmd = "python {} {} {} {}".format(reader, reader_class, "TRAIN", self._config_yaml) else: