mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-09 06:05:11 -05:00
03b38864db
* e2e testing * min failure * no affine on bn, still fails * why did i think i could detach that? * allow more kernels for bn * some test issue i don't understand
14 lines
491 B
Python
14 lines
491 B
Python
#!/usr/bin/env python3
|
|
from examples.hlb_cifar10 import SpeedyResNet, fetch_batch
|
|
from examples.hlb_cifar10_torch import SpeedyResNet as SpeedyResNetTorch
|
|
from datasets import fetch_cifar
|
|
from test.models.test_end2end import compare_tiny_torch
|
|
|
|
if __name__ == "__main__":
|
|
X_test, Y_test = fetch_cifar(train=False)
|
|
X, Y = fetch_batch(X_test, Y_test, 32)
|
|
print(X.shape, Y.shape)
|
|
model = SpeedyResNet()
|
|
model_torch = SpeedyResNetTorch()
|
|
compare_tiny_torch(model, model_torch, X, Y)
|