mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-07 03:00:26 -04:00
move good models to models, add a training step test
This commit is contained in:
@@ -10,7 +10,7 @@ import numpy as np
|
||||
np.set_printoptions(suppress=True)
|
||||
from tinygrad.tensor import Tensor
|
||||
from extra.utils import fetch, get_parameters
|
||||
from extra.efficientnet import EfficientNet
|
||||
from models.efficientnet import EfficientNet
|
||||
|
||||
def infer(model, img):
|
||||
# preprocess image
|
||||
|
||||
Reference in New Issue
Block a user