20 commits

Author	SHA1	Message	Date
George Hotz	718ced296c	move state to nn/state (#1619)	2023-08-22 07:36:24 -07:00
Rob Grossman	2533a992e7	remove unused imports in models (#1088)	2023-07-01 12:04:19 -07:00
George Hotz	791530045d	Refactor LoadOps (#910) ... * test * work * upd test * loadops * cleanups * real ones * remove LazyNumpyArray * fix assign test * remove range * np.require * llama uses arange kernels * no caching consts * fix enet * torch load support * tests cleanup * fix shufflenet * fix image * fix torch_load test	2023-06-03 09:40:43 -07:00
George Hotz	b1206bcb18	third try at torch loading (#677) ... * third try at torch loading * numpy fixed * fix enet compile * load_single_weight supports empty weights * oops, CPU wasn't the default * so many bugs	2023-03-10 19:11:29 -08:00
Jacky Lee	e172f0087a	BatchNorm2D -> BatchNorm2d (#558) ... * BatchNorm2D -> BatchNorm2d * Fix typo	2023-02-16 12:31:49 -08:00
George Hotz	a0d169eb59	fix efficientnet	2022-09-28 14:23:01 -07:00
Comma Device	a734df98fa	TEST_ENET for openpilot compiler	2022-08-31 13:23:36 -04:00
George Hotz	368c0ce2f6	NUM=-2 for ants	2022-07-02 15:47:10 -07:00
George Hotz	0cb99d72e9	NUM=-1 is a small efficientnet for small people	2022-07-02 15:11:51 -07:00
George Hotz	8cf1aed0f4	don't track_running_stats, parameters must require_grad	2022-07-02 14:38:45 -07:00
George Hotz	67ff6b52fd	move padding to convs in enet	2022-06-26 23:14:31 -07:00
George Hotz	892ac661e1	enet readability	2022-06-07 10:23:05 -07:00
George Hotz	e28cdfb0cf	clean up resnet	2021-11-30 16:14:54 -05:00
George Hotz	8f5779eeaa	very minor change	2021-11-30 15:54:03 -05:00
George Hotz	46bbbcf7f0	model touchups	2021-11-30 11:13:34 -05:00
George Hotz	908db3bdea	support bias in conv like linear	2021-11-30 00:44:59 -05:00
George Hotz	bd21304e3c	linear takes in weight and bias	2021-11-30 00:38:47 -05:00
George Hotz	7d12482d80	refactor efficientnet loading	2021-10-30 17:02:17 -07:00
Sebastian Kreft	3358770182	chore(efficientnet): don't use eval when loading weights (#286) ... Because the weights are being loaded from a third party internet address, it's unsafe to use eval. Also with the change I think the code became a little bit more clear as now it's clearer which keys are being transformed. Co-authored-by: Seba Kreft <sebastian.kreft@houm.com>	2021-10-22 15:10:04 -07:00
George Hotz	d3f169b267	move good models to models, add a training step test	2021-06-19 11:24:15 -07:00

Renamed from extra/efficientnet.py (Browse further)