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