deepcrayon
/
tinygrab
1
0
Fork 0
Code

update todo

pull/269/head
George Hotz 2021-06-15 10:22:39 -07:00
parent 4e1edb3692
commit 4850d6eb43
2 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
README.md
View File

@ -153,12 +153,9 @@ tinygrad will always be below 1000 lines. If it isn't, we will revert commits un
python3 -m pytest
```
### TODO
### TODO (updated)
* Add reduce instructions to RISK
* Make EfficientNet backward pass fully use RISK instructions
* Train an EfficientNet on ImageNet
* Add a language model. BERT?
* Add a detection model. EfficientDet?
* Reduce code
* Increase speed
* Add features

2
extra/risk.py
View File

@ -168,7 +168,7 @@ load_log = open("/tmp/risk_load_log", "w") if os.getenv("LOAD_LOG") else None
def riski_load(target, address, stride_y=SZ, stride_x=1, len_y=SZ, len_x=SZ):
global util_n, util_d
if load_log is not None:
load_log.write("%d %d %d\n" % (address, stride_y, stride_x))
load_log.write("%d %d %d %d %d\n" % (address, stride_y, stride_x, len_y, len_x))
utils[(len_y, len_x)] += 1
stride_y, stride_x = int(stride_y), int(stride_x)
d = regfile[target]