mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases 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
parent ee99d016e9
commit 5797e63d9b
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