feat(ml): jenkins test inference print

jolibrain · Sep 17, 2024 · b68ab53 · b68ab53
1 parent 70919cd
commit b68ab53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/palette_model.py b/models/palette_model.py
@@ -648,6 +648,7 @@ def inference(self, nb_imgs, offset=0):
 
         # other tasks
         else:
+            print("canny task inference nb_imgs , sample_num", nb_imgs, self.sample_num)
             self.output, self.visuals = netG.restoration(
                 y_cond=self.cond_image[:nb_imgs], sample_num=self.sample_num
             )