github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

model touchups

Browse Source
This commit is contained in:
George Hotz
2021-11-30 11:13:34 -05:00
parent 7d7e2b690d
commit 46bbbcf7f0
5 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/efficientnet.py
View File

@@ -49,7 +49,7 @@ def infer(model, img):
if __name__ == "__main__":
# instantiate my net
model = EfficientNet(int(os.getenv("NUM", "0")))
model.load_weights_from_torch()
model.load_from_pretrained()
# category labels
import ast