Commit Graph
13471 Commits
Author SHA1 Message Date
geohot 4e1a0de392 fix rsub 2020-12-08 10:05:21 -08:00
geohot c4540f1b8c Support scalars by kartik4949 2020-12-08 09:52:07 -08:00
geohot 97fd9c1237 zero_grad there to match readme 2020-12-07 23:12:18 -08:00
geohot c63f950348 need zero grad now 2020-12-07 23:10:43 -08:00
George HotzandGitHub b355cd2571 Mean axis (doesn't work) (#154)
* mean axis

* fixed
2020-12-07 22:58:34 -08:00
geohot 38f97c8c80 prepare for ops_ane 2020-12-07 21:54:22 -08:00
geohot 7f249ec76d touch up 2020-12-07 21:51:32 -08:00
Marcel BischoffandGitHub 58ccebd7cd Sum with axis (#153)
* sum with axis and tests

* broken

* works again

* clean up

* Update test_ops.py
2020-12-07 21:49:18 -08:00
geohot ac9fecb05d lots of notes 2020-12-07 21:40:31 -08:00
geohot 8d1500f497 conv neuron 2020-12-07 21:12:52 -08:00
geohot e4bb53b0e9 work out more 2020-12-07 20:32:50 -08:00
geohot 4927ad1897 float16 weights in min.weights 2020-12-07 20:15:15 -08:00
geohot 3aac9aefce fix GPU profiling 2020-12-07 20:03:28 -08:00
James RobertsandGitHub b2eca6d45f Format debug output (#152) 2020-12-07 14:07:14 -08:00
geohot c7973cb0a1 ugh buffer_np is bad 2020-12-07 08:07:00 -08:00
geohot 088f280dc3 touchups 2020-12-07 07:50:27 -08:00
geohot 0cf21881b7 hwx parse w/o macho mods 2020-12-06 23:13:28 -08:00
Josh SmithandGitHub aa4161f63e use classmethods for Tensor helper funcs (#146) 2020-12-06 22:35:43 -08:00
geohot e75a6d1b4b quadconv 2020-12-06 20:39:50 -08:00
geohot 23664c99bd double conv 2020-12-06 20:26:02 -08:00
geohot 1a0f826dc6 highlight the commands 2020-12-06 20:03:21 -08:00
geohot 2f1f006003 we have docs 2020-12-06 19:54:03 -08:00
geohot 3531e81f0e dumping ANE docs 2020-12-06 18:58:36 -08:00
geohot 6e793e96c3 deeebug 2020-12-06 17:49:17 -08:00
geohot dced0cb44b oops, path to weights 2020-12-06 16:33:42 -08:00
geohot c57dc61ea7 simple op examples 2020-12-06 16:32:26 -08:00
geohot 7babf38617 found concat 2020-12-06 16:27:12 -08:00
geohot ddd6778423 add neuron 2020-12-06 16:24:42 -08:00
geohot f2f2d6aea3 docs and noop 2020-12-06 16:10:44 -08:00
geohot 6ba25834ee found some plists 2020-12-06 15:57:28 -08:00
geohot e2184c20ad min weights, update golden 2020-12-06 15:29:15 -08:00
geohot d4d8bd0337 make minimal plist for compare 2020-12-06 15:10:15 -08:00
geohot 0845ec43c6 compile takes in plist 2020-12-06 14:51:33 -08:00
geohot 00312b8ad1 batchnorm work 2020-12-06 14:40:07 -08:00
geohot da514c2918 fix enet init 2020-12-06 13:52:07 -08:00
geohot 3b982f2f7a get_parameters 2020-12-06 13:47:28 -08:00
geohot 102e6356e9 replace layer_init_uniform with .uniform 2020-12-06 13:44:31 -08:00
geohot c71a8ef222 remove unused Div op 2020-12-06 13:02:12 -08:00
geohot 20f95de408 less lines 2020-12-06 12:56:03 -08:00
geohot 629efb391f lose a few lines 2020-12-06 12:42:41 -08:00
geohot 521098cc2f se optional, track time better 2020-12-06 12:29:42 -08:00
geohot 609d11e699 trainer works with CIFAR 2020-12-06 12:20:14 -08:00
geohot 80a9c777ba requires grad, optim in train enet 2020-12-06 11:10:30 -08:00
geohot c66c27d22e get parameters 2020-12-06 10:45:04 -08:00
geohot 51daaa43d4 fix memory leaks, add gc test 2020-12-06 10:34:40 -08:00
geohot 1717daa859 reshape doesn't copy anymore 2020-12-06 09:51:09 -08:00
geohot 62ee47fef8 add GPUBuffer class 2020-12-06 09:45:13 -08:00
geohot 17659f7dd7 gpu speedup, tests work on M1 2020-12-06 09:05:49 -08:00
geohot b8deb36e56 train BS=16 for 32 steps 2020-12-04 10:00:32 -08:00
geohot ad1b225722 oops, i broke fill 2020-12-04 09:53:38 -08:00