Skip to content

Commit f00d26a

Browse files
committed
Add readme
1 parent fefe3cc commit f00d26a

File tree

2 files changed

+18
-2
lines changed

2 files changed

+18
-2
lines changed

README.md

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
# Intro
2+
3+
PyTorch implementation of [Learning to learn by gradient descent by gradient descent](https://arxiv.org/abs/1606.04474).
4+
5+
## Run
6+
7+
```bash
8+
python main.py
9+
```
10+
11+
### TODO
12+
- [x] Initial implementation
13+
- [x] Toy data
14+
- [x] LSTM updates
15+
- [ ] Compare with standard optimizers
16+
- [ ] Real data
17+
- [ ] More difficult models

main.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,7 @@
4141
# Reset lstm values of the meta optimizer
4242
meta_optimizer.reset_lstm()
4343

44-
x, y = get_batch(args.batch_size
45-
)
44+
x, y = get_batch(args.batch_size)
4645
x, y = Variable(x), Variable(y)
4746

4847
# Compute initial loss of the model

0 commit comments

Comments
 (0)