X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=quizz_machine.py;h=697f27ece4e353fe8a264657013a7fe2f693a630;hb=c9c018e4c19ce92892d7652082fb90719d57441c;hp=4e7576e720ffa9872701e6174c07667bd1c782c8;hpb=98dbe305561906ad65deb5245aa7aeeb7a824fb2;p=culture.git diff --git a/quizz_machine.py b/quizz_machine.py index 4e7576e..697f27e 100755 --- a/quizz_machine.py +++ b/quizz_machine.py @@ -139,6 +139,7 @@ class QuizzMachine: self.train_w_quizzes = self.problem.generate_token_sequences( nb_train_samples ).to(device) + self.test_w_quizzes = self.problem.generate_token_sequences(nb_test_samples).to( device )