mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-24 02:14:17 +00:00
2793cca9a6* add support for a custom BASEDIR for openimages download * make export step faster * add focal loss * update model_eval with new dataloader * generate_anchors in tinygrad * update initializers for model * small cleanup * revert isin enhancements * recursively go through backbone layers to freeze them * add optimizer * minor cleanup * start dataloader work with input images * add first transform for train set * reuse existing prepare_target * continue with dataloader implementation * add dataloader * separate out KiTS19 dataset test cases * create mock data samples for test * add dataloader + test * cleanup dataloader test and revert shm path * trim dataloader related code needed from ref * got dataloader with normalize working * update image to be float32 * add back normalization and negate it in test * clean up reference dataset implementation + ruff changes * add validation set test * add proper training loop over the training dataset * add LambdaLR support * add LR scheduler and the start of training step * get forward call to model work and setup multi-GPU * already passed device * return matches from dataloader * hotfix for dataloader typo causing some hang * start some work on classification loss * update focal loss to support masking * add missing test and cleanup focal loss * cleanup unit tests * remove masking support for sigmoid_focal_loss * make ClassificationHead loss work * cleanups + fix dataloader tests * remove sigmoid when computing loss * make anchors use Tensors * simplify anchors batching * revert anchors to use np * implement regression loss * fix regression loss * cleanup losses * move BoxCoder to MLPerf helpers * revert helper changes * fixes after helper refactor cleanup * add tests for l1_loss * start re-enabling training step * minor cleanup * add pycocotools to testing dependencies * make training work * adjust regression loss to mask after L1 loss is calculated * reduce img and lbl sizes by half for KiTS19 dataset tests * Revert "reduce img and lbl sizes by half for KiTS19 dataset tests" This reverts commit |
||
|---|---|---|
| .. | ||
| conversation_data | ||
| llm.c | ||
| mlperf | ||
| openpilot | ||
| other_mnist | ||
| rl | ||
| sovits_helpers | ||
| tinychat | ||
| vgg7_helpers | ||
| webgpu | ||
| __init__.py | ||
| beautiful_cartpole.py | ||
| beautiful_cifar.py | ||
| beautiful_mnist.py | ||
| beautiful_mnist_multigpu.py | ||
| benchmark_onnx.py | ||
| coder.py | ||
| compile_efficientnet.py | ||
| compile_tensorflow.py | ||
| conversation.py | ||
| efficientnet.py | ||
| flux1.py | ||
| flux1_seed0.png | ||
| gpt2.py | ||
| handcode_opt.py | ||
| hlb_cifar10.py | ||
| llama.py | ||
| llama3.py | ||
| mamba.py | ||
| mask_rcnn.py | ||
| mixtral.py | ||
| mnist_gan.py | ||
| olmoe.py | ||
| openelm.py | ||
| qwq.py | ||
| sdv2.py | ||
| sdxl.py | ||
| sdxl_seed0.png | ||
| self_tokenize.py | ||
| serious_mnist.py | ||
| simple_conv_bn.py | ||
| so_vits_svc.py | ||
| stable_diffusion.py | ||
| stable_diffusion_seed0.png | ||
| stunning_mnist.py | ||
| test_onnx_imagenet.py | ||
| test_pkl_imagenet.py | ||
| torch_cuda_kernel.py | ||
| train_efficientnet.py | ||
| train_resnet.py | ||
| transformer.py | ||
| vgg7.py | ||
| vit.py | ||
| vits.py | ||
| whisper.py | ||
| yolov3.py | ||
| yolov8-onnx.py | ||
| yolov8.py | ||
