diff --git a/cluster/train_cluster.py b/cluster/train_cluster.py index 7e42de4..dfa55e6 100644 --- a/cluster/train_cluster.py +++ b/cluster/train_cluster.py @@ -14,6 +14,9 @@ logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) def train_cluster(in_dir, n_clusters, use_minibatch=True, verbose=False,use_gpu=False):#gpu_minibatch真拉,虽然库支持但是也不考虑 + if in_dir.endswith(".ipynb_checkpoints"): + logger.info(f"Ignore {in_dir}") + logger.info(f"Loading features from {in_dir}") features = [] nums = 0