projects
/
culture.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[culture.git]
/
quizz_machine.py
diff --git
a/quizz_machine.py
b/quizz_machine.py
index
806dde7
..
6f7492d
100755
(executable)
--- a/
quizz_machine.py
+++ b/
quizz_machine.py
@@
-386,8
+386,11
@@
class QuizzMachine:
ar_mask_solve = 1 - ar_mask_prompt
seq_logproba = torch.empty(ar_mask_prompt.size(0), device=self.device)
ar_mask_solve = 1 - ar_mask_prompt
seq_logproba = torch.empty(ar_mask_prompt.size(0), device=self.device)
- warnings.warn("very high temperature with reversed cleanup", RuntimeWarning)
- temperature = 10
+ if reverse_cleanup:
+ warnings.warn("very high temperature with reversed cleanup", RuntimeWarning)
+ temperature = 10.0
+ else:
+ temperature = 1.0
# warnings.warn("noise injection", RuntimeWarning)
# noise_std = torch.rand(1).item()
# warnings.warn("noise injection", RuntimeWarning)
# noise_std = torch.rand(1).item()