14 ops to write for GPU

2026-01-08 14:43:57 -05:00 · 2020-10-31 10:59:30 -07:00
parent 06928cf3cc
commit e01e35e545
3 changed files with 35 additions and 28 deletions
--- a/README.md
+++ b/README.md
@@ -88,8 +88,10 @@ python -m pytest

 ### TODO

+* Train an EfficientNet
+  * EfficientNet backward pass
+  * Tensors on GPU (GPU support, must support Mac)
 * Reduce code
 * Increase speed
 * Add features
-* In that order