Set W for NLS training according to paper

henry-yeh · Dec 21, 2023 · 8bf1cd8 · 8bf1cd8
1 parent 12c9597
commit 8bf1cd8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tsp_nls/train.py b/tsp_nls/train.py
@@ -9,8 +9,8 @@
 
 
 EPS = 1e-10
-T=5
-W=0.95
+T = 5
+W = 0.95
 
 def train_instance(model, optimizer, data, n_ants):
     model.train()