tinygrad/test
Yixiang Gao 6480a1a180
CIFAR 94.03% (#1340)
* add disk_tensor

* fix jit

* new baseline before whitening

* whitening through torch

* whiting done currently at 91.65%

* 91.99%

* clean up mixup and 92.3%

* clean up 92.30%

* 92.49% before searching for new hyper-parameters

* fix CI

* fix white space

* add whitening init in test

* refactor, update hyperpara, 92.72%

* converting whiting to tinygrad operation

* update CI kernels count for CIFAR

* add pad reflect

* add random crop 92.53%

* update hyperpara 93%

* 93.15% on docker container, need to refactor the assignment for hyper param

* print out weights and bias to be separated

* bias/non-bias params separated

* fix whitespace

* clean up

* refactor hyper-param with dict

* refactor lr schedular params

* fix whitespace

* fix cross entropy loss

* fix whitespace

* move opt hyp to hyp dict

* minor fixup

* adjust model, loss scaling

* 92.74% while using half of compute as before

* update hyp for cutmix

* random shuffle during batches

* clean up

* updating the model

* update ConvGroup

* disable gradients for batchnorm layer weights

* whitespace

* 93.92%

* clean up

* finally 94%git add .!

* rewrite whitening to remove dependency on torch

* whitespace

* remove dependency on torch, 93.91%

* back to 94.03%

* clean up

* update test_real_world
2023-08-08 15:13:24 -07:00
..
external use fake buffer for external_test_speed_llama.py (#1478) 2023-08-07 22:05:44 -04:00
extra Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00
models CIFAR 94.03% (#1340) 2023-08-08 15:13:24 -07:00
unit move renamer to linearizer (#1442) 2023-08-05 08:53:25 -07:00
__init__.py All devices are equal! (#196) 2020-12-15 23:44:08 -08:00
Dockerfile Docker fix (#1039) 2023-06-25 10:38:58 -07:00
test_assign.py Limit dims based on max size (#1390) 2023-07-31 19:18:19 -07:00
test_conv.py Limit dims based on max size (#1390) 2023-07-31 19:18:19 -07:00
test_conv_shapetracker.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_custom_function.py flake8 (#1323) 2023-07-24 11:19:58 -04:00
test_dtype.py Add WEBGPU tests to CI (#1463) 2023-08-06 10:32:01 -07:00
test_gc.py fix up eye and fix gc test 2023-02-27 06:53:18 -08:00
test_helpers.py fix: Context (#1430) 2023-08-04 10:53:48 -04:00
test_jit.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_lazybuffer.py just cmplt (#1493) 2023-08-08 13:58:10 -07:00
test_linearizer.py dedup kernel args v2 (#1272) 2023-07-18 20:03:42 -07:00
test_net_speed.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_nn.py Limit dims based on max size (#1390) 2023-07-31 19:18:19 -07:00
test_ops.py just cmplt (#1493) 2023-08-08 13:58:10 -07:00
test_optim.py Add WEBGPU tests to CI (#1463) 2023-08-06 10:32:01 -07:00
test_randomness.py rng hlops: add normal and kaiming_normal (#1378) 2023-07-31 10:37:02 -07:00
test_specific_conv.py CI < 5 minutes (#1252) 2023-07-23 13:00:56 -07:00
test_speed_v_torch.py Add WEBGPU tests to CI (#1463) 2023-08-06 10:32:01 -07:00
test_symbolic_shapetracker.py Support constant expand to symbolic shape (#1411) 2023-08-02 21:21:22 -07:00
test_tensor.py Add WEBGPU tests to CI (#1463) 2023-08-06 10:32:01 -07:00
test_uops.py just cmplt (#1493) 2023-08-08 13:58:10 -07:00
test_webgpu.js Revert "update editorconfig, enforce via CI (#1343)" (#1380) 2023-07-31 10:35:50 -07:00