Commit Graph

2686 Commits

Author SHA1 Message Date
George Hotz b91fd3afad maxpool 2020-10-25 07:43:34 -07:00
George Hotz 5216a1d9f3 refactor into tensor and ops 2020-10-23 10:34:21 -07:00
George Hotz 9b9e47f369 added conv profile test 2020-10-23 09:46:10 -07:00
George Hotz 5756115e57 anyone else let down by the fast conv? 2020-10-23 09:09:29 -07:00
George Hotz bcb60e0b7c wow, you have to name them test 2020-10-23 06:33:18 -07:00
George Hotz 2259c9faa1 low lr improves rmsprop 2020-10-23 06:22:32 -07:00
George Hotz eda29fa0e0 clean up test 2020-10-23 06:11:38 -07:00
George Hotz 373b4e341b
Merge pull request #15 from f0ti/master
added RMSprop optim
2020-10-23 06:08:20 -07:00
f0ti 0b87aaca1e update rsmprop 2020-10-23 14:46:45 +02:00
f0ti c5f726ec2e all three 2020-10-23 11:53:01 +02:00
f0ti 6a38ccb6b0 update rmsprop and readme 2020-10-23 11:49:43 +02:00
George Hotz 21ebb0b769 if you wait 24 seconds, that gets 98% 2020-10-22 21:49:14 -07:00
George Hotz 816f648161 chans doesn't need to be in self 2020-10-22 21:19:35 -07:00
George Hotz 77251cc6c3 7x7 conv = more accuracy 2020-10-22 21:10:27 -07:00
f0ti 7e1eddb0c5 added RMSprop optim 2020-10-23 02:50:02 +02:00
0xNaN d95adbddb4 `gradcheck` now returns only a bool, refactoring of test_gradcheck 2020-10-22 01:28:52 +02:00
0xNaN adbfc67456 test `jacobian` and `numerical_jacobian` against torch.autograd.functional.jacobian 2020-10-22 01:28:52 +02:00
0xNaN 1561d3b9c0 extracting `jacobian` and `test_jacobian` 2020-10-22 01:28:52 +02:00
0xNaN 93bc3c22a0 tiny gradcheck 2020-10-22 01:28:52 +02:00
Adrian Garcia Badaracco 9a8be135a7
incorporate changes 2020-10-21 13:21:44 -05:00
Adrian Garcia Badaracco 02adb0ac3a
Make test_mnist runnable by pytest and directly 2020-10-21 11:30:08 -05:00
Adrian Garcia Badaracco 5afe6b1f68
rename files 2020-10-21 11:28:03 -05:00
George Hotz d91902948b add reshape support and OMG the CONVS are SO SLOW 2020-10-21 09:12:19 -07:00
George Hotz e3110c9922 backward pass for conv2d, lol i mostly guessed and made shapes match 2020-10-21 08:45:35 -07:00
George Hotz 5c2ac48c11 write forward pass for convolution 2020-10-19 09:33:06 -07:00
George Hotz 2681c79bc5 simple tests, repr not str 2020-10-18 14:55:20 -07:00
George Hotz 4019c38942 more readme 2020-10-18 14:38:20 -07:00
George Hotz cc9054e3ec refactor into utils 2020-10-18 14:36:29 -07:00
George Hotz 0c3dd12b3b i hate tabs 2020-10-18 14:33:13 -07:00
George Hotz a139f34bb6 fix nll loss in example 2020-10-18 14:27:54 -07:00
George Hotz 26ce2d93c3 add support for adam 2020-10-18 13:50:23 -07:00
George Hotz 6532233d24 refactor better 2020-10-18 13:33:02 -07:00
George Hotz 92fd23df66 refactor into a few files 2020-10-18 13:30:25 -07:00
George Hotz 118c2eebe3 write sgd class 2020-10-18 13:27:59 -07:00
George Hotz 54eafe6c12 update readme 2020-10-18 13:08:14 -07:00
George Hotz 83417d4b4c readme and dirs 2020-10-18 12:48:17 -07:00