mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-15 17:15:48 -05:00
aa60feda48
* Rename in files * Move files * Moved to extra/datasets as suggested * Changes to files * Fixed stupid mistake --------- Co-authored-by: terafo <terafo@protonmail.com>
14 lines
497 B
Python
14 lines
497 B
Python
#!/usr/bin/env python3
|
|
from examples.hlb_cifar10 import SpeedyResNet, fetch_batch
|
|
from examples.hlb_cifar10_torch import SpeedyResNet as SpeedyResNetTorch
|
|
from extra.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)
|