mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

Train efficientnet should respect NUM environment variable (#122)

Browse source 
Co-authored-by: holonomicjl <58403584+holonomicjl@users.noreply.github.com>
This commit is contained in:
adamritter 2020-11-17 04:02:31 +00:00 • committed by GitHub
commit 5797e63d9b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/train_efficientnet.py
View file

@ -6,7 +6,7 @@ from tinygrad.tensor import Tensor
if __name__ == "__main__":
Tensor.default_gpu = os.getenv("GPU") is not None
model = EfficientNet()
model = EfficientNet(int(os.getenv("NUM", "0")))
BS = 4