move globalcounters to ops (#2960)

* move globalcounters to ops

* missed a few

* sick of that failing
This commit is contained in:
George Hotz
2024-01-01 14:21:02 -08:00
committed by GitHub
parent 8291986959
commit c81ce9643d
23 changed files with 52 additions and 282 deletions

View File

@@ -17,9 +17,8 @@ from extra.datasets import fetch_cifar, cifar_mean, cifar_std
from tinygrad import nn
from tinygrad.nn.state import get_state_dict
from tinygrad.nn import optim
from tinygrad import Device
from tinygrad import Device, GlobalCounters
from tinygrad.tensor import Tensor
from tinygrad.helpers import GlobalCounters
from tinygrad.shape.symbolic import Node
from extra.lr_scheduler import OneCycleLR
from tinygrad.jit import TinyJit