128 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
George Hotz	43591a1e71	make the example simpler	2020-10-26 09:19:20 -07:00
George Hotz	1f0514e5df	pip	2020-10-26 09:15:31 -07:00
George Hotz	64bd4f7936	lol, it's not 1.0	2020-10-26 09:11:32 -07:00
George Hotz	dec2f1c31a	Merge pull request #18 from goktug97/master ... add setup.py and change imports to relative	2020-10-26 09:09:31 -07:00
Göktuğ Karakaşlı	81040d926d	import only the module	2020-10-26 19:02:06 +03:00
Göktuğ Karakaşlı	8d80726207	two spaces	2020-10-26 18:54:55 +03:00
George Hotz	1654008c1f	conv stride support	2020-10-26 08:54:43 -07:00
George Hotz	2a55d7402b	clean up ops, refactor pool backward. add stride test	2020-10-26 08:47:11 -07:00
George Hotz	93dceb4bee	fix kernel_size bug, name like torch, add test	2020-10-26 08:38:53 -07:00
Göktuğ Karakaşlı	cc9bd45b44	add setup.py and change imports to relative	2020-10-26 18:19:50 +03:00
George Hotz	d0c9be3e25	Merge pull request #17 from tsmcalister/master ... makes maxpool2d parameterizable	2020-10-26 06:35:34 -07:00
Timothy Mc Alister	15e5988323	make default parameters work for functions	2020-10-26 12:43:36 +01:00
George Hotz	2d37fd686b	test ops	2020-10-25 19:03:49 -07:00
George Hotz	2eebbd32c6	ops test speed	2020-10-25 19:01:02 -07:00
George Hotz	b27bcbe4b4	avgpool and test refactor	2020-10-25 18:40:01 -07:00
George Hotz	4c42676cb6	400 -> 200	2020-10-25 17:19:59 -07:00
George Hotz	567707a5f6	rename max_pool2d to match torch, remove more fast conv crap	2020-10-25 17:16:47 -07:00
George Hotz	5d1373c71b	if you want fast convs, revert this	2020-10-25 17:10:34 -07:00
George Hotz	ea41f5e1c1	seems more generic	2020-10-25 16:40:37 -07:00
George Hotz	2333c4dea7	no tqdm in actions	2020-10-25 16:40:08 -07:00
George Hotz	ad48061927	better sort in torch profiler	2020-10-25 16:07:49 -07:00
George Hotz	82f8e10813	no hacks in that test	2020-10-25 15:52:05 -07:00
George Hotz	4baa4c041f	it's crazy how much faster pytorch is than numpy	2020-10-25 15:42:33 -07:00
George Hotz	5ddbd7f04b	2 to 3x slower than torch	2020-10-25 15:27:33 -07:00
George Hotz	f8311f5ecd	print fp/bp mnist	2020-10-25 15:08:18 -07:00
George Hotz	d38367b561	cleaner relu	2020-10-25 14:35:04 -07:00
George Hotz	5c179d18ad	add profiling for mnist net	2020-10-25 14:20:55 -07:00
George Hotz	8fcada8071	faster and better convnet	2020-10-25 13:48:44 -07:00
George Hotz	1255903519	hmm, no reason it should have been wrong, numpy weirdness	2020-10-25 13:14:12 -07:00
George Hotz	96f9cdb8a0	woah, fastconv is wrong	2020-10-25 12:56:42 -07:00
George Hotz	bb98cdfef7	improve conv testing	2020-10-25 12:46:04 -07:00
George Hotz	ef24aac09e	finally, fast convs	2020-10-25 12:39:44 -07:00
George Hotz	0f02084805	conv isn't fast yet	2020-10-25 12:13:58 -07:00
George Hotz	12641d5bd7	umm, okay slow col2im	2020-10-25 12:01:59 -07:00
George Hotz	67506eb6ba	fast im2col	2020-10-25 11:49:35 -07:00
George Hotz	c9968756d1	allow the line profiler to work	2020-10-25 11:13:40 -07:00
George Hotz	5062c2c8ff	profile conv better	2020-10-25 11:11:00 -07:00
George Hotz	c74764bac3	oops, set to None	2020-10-25 08:28:18 -07:00
George Hotz	935f5ddaaa	always keep batch size out front	2020-10-25 08:14:07 -07:00
George Hotz	b91fd3afad	maxpool	2020-10-25 07:43:34 -07:00
George Hotz	3cef527e80	Merge pull request #14 from adriangb/reqs-txt ... requirements.txt	2020-10-24 07:06:08 -07:00
Adrian Garcia Badaracco	02ecdc86c3	remove install from workflow	2020-10-24 09:03:22 -05:00
Adrian Garcia Badaracco	2a64bd5980	everyone's a dev	2020-10-24 09:01:15 -05: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	f16c5577f0	useless parens	2020-10-23 06:29:57 -07:00
George Hotz	0f358a9cfb	remove the eps from the sqrt	2020-10-23 06:28:32 -07:00
George Hotz	2259c9faa1	low lr improves rmsprop	2020-10-23 06:22:32 -07:00