unfinished work on backtracking

sudoku/train.py

@@ -401,6 +401,9 @@ class SudokuLightning(pl.LightningModule):
 
 # TODO adapt training to something softer
 # 
+class TrialEveryPosException(Exception):
+    pass
+
 class SudokuTrialErrorLightning(SudokuLightning):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
@@ -752,7 +755,7 @@ class SudokuTrialErrorLightning(SudokuLightning):
             mask_possibility[pos]=False
         if  mask_possibility.sum()==0:
             print('mask_possible=0')
-            raise ValueError()
+            raise TrialEveryPosException()
             
         with torch.no_grad():
             x_reg = self.sym_preprocess.forward(s_new_X.view(1,2,-1))
@@ -861,7 +864,28 @@ class SudokuTrialErrorLightning(SudokuLightning):
                         return x
                     X_tried = new_X
                     # if one of X_tried is complete (weird but possible) -> return x with tried_position mask_complet set to 1 (cause we still want a step by step resolution)
+    
+    def backtracking_predict(self, x):
+        """
+        return is_valid, new_x
+        """
+        try:
+            x = self.predict(x)
+        except TrialEveryPosException:
+            pos = self.search_trial(x.view(2,729), [])
+            new_x = deepcopy(x.view(2,729))
+            output = self.forward_layer(x.view(1,2,729))
+            pos_output = torch.argmax(output[0,:,pos])
+            if (pos_output[0].item()-self.threshold_abs)>(pos_output[1].item-self.threshold_pos):
+                trial_abs_pres = 0
+            else:
+                trial_abs_pres = 1
+            new_x[0,trial_abs_pres,pos]=1
+            is_valid, new_x = self.backtracking_predict(new_x)
+            if not is_valid:
+                return is_valid ### TO continue
             
+    
     def on_validation_epoch_start(self) -> None:
         # self.buffer = BufferArray(self.nets_number, self.batch_size)
         self.trial_error_buffer = Buffer(self.batch_size)