projects
/
beaver.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (from parent 1:
1e6f089
)
Update
author
François Fleuret
<francois@fleuret.org>
Tue, 14 Mar 2023 20:05:11 +0000
(21:05 +0100)
committer
François Fleuret
<francois@fleuret.org>
Tue, 14 Mar 2023 20:05:11 +0000
(21:05 +0100)
beaver.py
patch
|
blob
|
history
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(),
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):
).to(device)
for n_epoch in range(args.nb_epochs):