diff --git a/benchmark/alexnet/report.md b/benchmark/alexnet/report.md
index ca7ef7df59359a79960a8273b989757b1481216e..946cd9cc1438e53ddbdc849202e5ab855760e06c 100644
--- a/benchmark/alexnet/report.md
+++ b/benchmark/alexnet/report.md
@@ -2,6 +2,6 @@ batch_size: 1024
gpu num | time (one batch)
:-------| :-------------
-1 | 538ms
-2 | 275ms
-4 | 214ms
+1 | 541ms
+2 | 282ms
+4 | 207ms
diff --git a/benchmark/alexnet/train.net b/benchmark/alexnet/train.net
index 3cdcd9b077c733fa05f580a1bb15a49505d04305..4e536fa186362733d79a632769c6c5e4d5fb70a8 100644
--- a/benchmark/alexnet/train.net
+++ b/benchmark/alexnet/train.net
@@ -1,7 +1,7 @@
op {
name: "decode"
decode_ofrecord_conf {
- data_dir: "/dataset/imagenet/train/32"
+ data_dir: "/dataset/imagenet_227/train/32/"
blob {
name: "encoded"
data_type: kFloat
@@ -12,12 +12,6 @@ op {
}
encode_case {
jpeg {
- preprocess {
- resize {
- width: 227
- height: 227
- }
- }
}
}
}
diff --git a/oneflow/of_submit b/oneflow/of_submit
index 8e83c2d326188373024da7606a4c97dc0c7e8cc1..abe648167d1d5ac77db92bf7631eac35d6a97164 100755
--- a/oneflow/of_submit
+++ b/oneflow/of_submit
@@ -55,6 +55,16 @@ def SubmitJob(job_conf, oneflow_path, workdir):
for machine in job_conf.resource.machine:
SubmitJobToOneMachine(machine, job_conf, oneflow_path, workdir)
+def FindOneflowPathInEnv():
+ for path_dir in os.getenv("PATH").split(":"):
+ try:
+ filenames = os.listdir(path_dir)
+ if "oneflow" in filenames:
+ return os.path.join(path_dir, "oneflow")
+ except os.error:
+ pass
+ return ""
+
if __name__ == "__main__":
if len(sys.argv) == 1:
PrintHelpMessage()
@@ -68,7 +78,9 @@ if __name__ == "__main__":
else:
PrintHelpMessage()
oneflow_path = os.path.abspath(os.path.expanduser(oneflow_path))
- assert os.path.isfile(oneflow_path)
+ if os.path.isfile(oneflow_path) == False:
+ oneflow_path = FindOneflowPathInEnv()
+ assert os.path.isfile(oneflow_path)
workdir = os.path.abspath(os.path.expanduser(workdir))
job_conf_path = sys.argv[-1]
if os.path.isfile(job_conf_path) == False: