move good models to models, add a training step test

This commit is contained in:
George Hotz
2021-06-19 11:24:15 -07:00
parent b48d4bad2e
commit d3f169b267
6 changed files with 55 additions and 6 deletions

View File

@@ -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