4,777 commits

Author	SHA1	Message	Date
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