1
0
Fork 0
tinygrab/.gitignore

14 lines
100 B
Plaintext
Raw Normal View History

2020-10-17 23:57:01 -06:00
__pycache__
notebooks
.*.swp
2020-11-07 11:46:36 -07:00
.*.swo
[WIP] YOLO working on tinygrad! (#245) * Some progress on yolov3 * Removed some debugging comments… Also, the forward pass eats all RAM for some reason * forward pass almost runs * forward pass runs almost * forward pass runs, now we gotta load the weights * loading weights works * fetches config and weights * everything kind of works, postprocessing of output still needs to be implemented, temp_process_results kind of works, but its kind of terrible, and not how things should be done * some changes * fixed some bugs in the forward pass and load_weights function, now outputs more correct values, however some values are still loaded incorrectly * Something is wrong with the forward pass, Conv2d tests added * forward pass almost outputs correct values, gotta fix one more thign * yolo works * some final changes * reverting changes * removed dataloader * fixed some indentation * comment out failing test, somehow it fails CI even though it passes on my computer… * fixed wrong probabilities * added webcam option to YOLO, now just need to add bounding boxes and speed it up * some progress towards adding bounding boxes * trying to speed up yolo layer on GPU, still faster on CPU but with 30GB ram usage * Faster inference times, bounding boxes added correctly, webcam works, but is slow, and there is a memory leak when running on CPU... Also added tinygrads output on the classic dog image * removed some debugging print statements * updated result image * something weird is going on, mean op on GPU tensor randomly faults, copying a tensor from GPU->CPU takes 10+ seconds…
2021-04-25 19:06:52 -06:00
*.pyc
*.so
build
dist
*.egg-info
2020-11-07 11:46:36 -07:00
/env
a.out
2021-10-30 20:40:44 -06:00
boxes.jpg
2022-09-06 09:06:11 -06:00
weights/*