github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Refactor getenv into helpers (#508)

Browse Source 
* Refactor getenv into helpers

* Remove unused os

* Fix default value

* Fix more defaults for CI

* Fix bracket

* Revert changes to openpilot/compile.py

* Use getenv from helpers when possible
This commit is contained in:
Jacky Lee
2023-01-31 15:09:09 -08:00
committed by GitHub
parent d91b6711ea
commit 799b3f185a
29 changed files with 102 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_train.py
View File

@@ -1,9 +1,9 @@
import os
import unittest
import time
import tinygrad.nn.optim as optim
import numpy as np
from tinygrad.tensor import Device
from tinygrad.helpers import getenv
from extra.training import train
from extra.utils import get_parameters
from models.efficientnet import EfficientNet
@@ -11,7 +11,7 @@ from models.transformer import Transformer
from models.vit import ViT
from models.resnet import ResNet18
BS = int(os.getenv("BS", "2"))
BS = getenv("BS", 2)
def train_one_step(model,X,Y):
params = get_parameters(model)