Train efficientnet should respect NUM environment variable
MODIFY
examples/train_efficientnet.py
examples/train_efficientnet.py
@@ -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

Train efficientnet should respect NUM environment variable (#122)

Co-authored-by: holonomicjl <58403584+holonomicjl@users.noreply.github.com>