6 commits

Author	SHA1	Message	Date
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	4d28d55683	add nn layer tests	2023-06-01 21:34:24 -07:00
George Hotz	8a928ed2f3	nn init matches torch (#901)	2023-06-01 21:24:11 -07:00
Rabia Eda Yılmaz	3075988468	Added kaiming_uniform initialization for Conv2d and Linear layers (#756) ... * added kaiming_uniform init for conv2d and linear layers * fix: set getattr * up * fix: set getattr * fix comments * better does not mean it is good * more nonlinearities * added test checks the distribution of default relu option * prettier * fix kernel size * edit distribution of returned tensor * complete tests and fix fan_mode * added higher dim test * prettier test * fix silly blank * just leaky_relu mode * default fan in and leaky relu * update params * fix test * shorter * generalize Tensor.uniform and adjust kaiming init - added low and high parameters to Tensor.uniform function, so it can have a specific range (default is 0 to 1) - adjusted return line of kaiming_uniform * range from -1 to 1 * delete comment * adjusted test_uniform * fixed * delete comment	2023-05-29 15:09:55 -07:00
Jacky Lee	b80cf9220c	Statistics test: check if distributions match torch (#769) ... * Check if tensor values match torch * Clean up randomness tests and remove dependency * Remove kaiming uniform test	2023-05-07 21:43:23 -07:00
Jacky Lee	5e41d5857c	Add tests for randomness (#621) ... * Add tests for random creation functions * It worked on my machine! * Rename to helper_same_distribution * Remove extra line * Add tests for equal distribution * Test without scipy * Do a different test for randn	2023-03-01 15:39:20 -08:00