projects
/
picoclvr.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[picoclvr.git]
/
tasks.py
diff --git
a/tasks.py
b/tasks.py
index
73f61bf
..
e7c2f75
100755
(executable)
--- a/
tasks.py
+++ b/
tasks.py
@@
-76,7
+76,7
@@
class Problem:
class ProblemLevel0(Problem):
def __init__(self, nb_sentences=100, len_prompt=5, len_result=5):
class ProblemLevel0(Problem):
def __init__(self, nb_sentences=100, len_prompt=5, len_result=5):
- self.seq = torch.randint(10, (nb_se
q
, len_prompt + 1 + len_result))
+ self.seq = torch.randint(10, (nb_se
ntences
, len_prompt + 1 + len_result))
self.seq[:, len_prompt] = 10
def generate_sequences(self, nb):
self.seq[:, len_prompt] = 10
def generate_sequences(self, nb):