Greeting! ML4CO-Kit 0.2.1

Author: heatingma

ml4co_kit/__init__.py

@@ -86,5 +86,5 @@
     from .learning.utils import points_to_distmat, sparse_points
 
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 __author__ = "SJTU-ReThinkLab"

ml4co_kit/algorithm/tsp/decoder/greedy.py

@@ -10,6 +10,7 @@ def tsp_greedy_decoder(heatmap: np.ndarray) -> np.ndarray:
     # check the number of dimension
     dim_2 = False
     if heatmap.ndim == 2:
+        dim_2 = True
         heatmap = np.expand_dims(heatmap, axis=0)
     if heatmap.ndim != 3:
         raise ValueError("``heatmap`` must be a 2D or 3D array.")

ml4co_kit/learning/train.py

@@ -70,7 +70,7 @@ def __init__(
         ckpt_save_path: Optional[str] = None,
         ckpt_monitor: str = "val/loss",
         save_top_k: int = -1,
-        mode: str = None,
+        mode: str = "min",
         ckpt_every_n_epochs: int = 1,
         ckpt_every_n_train_steps: Optional[int] = None,
         ckpt_filename: str = None,