train BS=16 for 32 steps
@@ -8,10 +8,12 @@ if __name__ == "__main__":
Tensor.default_gpu = os.getenv("GPU") is not None
model = EfficientNet(int(os.getenv("NUM", "0")))
- BS = 4
-
+ BS = 16
img = np.zeros((BS,3,224,224), dtype=np.float32)
+ for i in range(32):
+ print("running batch %d" % i)
+
st = time.time()
out = model.forward(Tensor(img))
et = time.time()
