X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=tasks.py;h=181ac4475a2126df33206e2a4cea29a3f2d57833;hb=732349f7c16e43ff84380d28e021d671f2c56492;hp=00b7a493481653d8eee786a5f3bf2299dacc3b01;hpb=a3f5378c684fb58b0dc839638e768c3a4b8e8a83;p=picoclvr.git

diff --git a/tasks.py b/tasks.py
index 00b7a49..181ac44 100755
--- a/tasks.py
+++ b/tasks.py
@@ -136,12 +136,9 @@ class TaskFromFile(Task):
                 pairs.append((sequence, pred_mask))
 
         symbols = ["#"] + list(set("".join([x[0] for x in pairs])) - set(["#"]))
-        print("SANITY", symbols)
         self.char2id = dict([(c, n) for n, c in enumerate(symbols)])
         self.id2char = dict([(n, c) for c, n in self.char2id.items()])
 
-        print(self.char2id)
-
         self.train_input, self.train_pred_masks = self.tensorize(
             pairs[:nb_train_samples]
         )
@@ -163,7 +160,6 @@ class TaskFromFile(Task):
         return len(self.char2id)
 
     def tensor2str(self, t):
-        print(f"{type(t)=}")
         return ["".join([self.id2char[x.item()] for x in s]) for s in t]
 
     def produce_results(