github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

resnet50 hand coded optimization (#1945)

Browse Source 
* resnet50 hand coded opt

* hand optimize one kernel

* opt in both places to fix test
This commit is contained in:
George Hotz
2023-09-29 09:34:51 -07:00
committed by GitHub
parent a677a1e2cd
commit 90326dbdc3
5 changed files with 90 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
models/resnet.py
View File

@@ -1,4 +1,5 @@
import tinygrad.nn as nn
from tinygrad.tensor import Tensor
from extra.utils import get_child
class BasicBlock:
@@ -114,7 +115,7 @@ class ResNet:
return out
return features
def __call__(self, x):
def __call__(self, x:Tensor) -> Tensor:
return self.forward(x)
def load_from_pretrained(self):