projects
/
beaver.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
| inline |
side by side
Update
[beaver.git]
/
beaver.py
diff --git
a/beaver.py
b/beaver.py
index
1408f0b
..
54510f0
100755
(executable)
--- a/
beaver.py
+++ b/
beaver.py
@@
-175,7
+175,7
@@
def one_shot(gpt, task):
model = nn.Sequential(
nn.Linear(args.dim_model, args.dim_model),
nn.ReLU(),
- nn.Linear(args.dim_model, 4)
+ nn.Linear(args.dim_model, 4)
,
).to(device)
for n_epoch in range(args.nb_epochs):