未验证 提交 06dd8fe8 编写于 作者: R ruri 提交者: GitHub

fix bugs in run script and utility.py (#2756)

上级 bc743ad2
......@@ -20,7 +20,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.1 \
......@@ -38,7 +37,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=piecewise_decay \
# --lr=0.1 \
......@@ -53,7 +51,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.1 \
......@@ -71,7 +68,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,299,299 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.045 \
......@@ -89,7 +85,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.1 \
......@@ -107,7 +102,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.1 \
......@@ -125,7 +119,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.1 \
......@@ -142,7 +135,6 @@ python train.py \
# --batch_size=256 \
# --total_images=1281167 \
# --image_shape=3,224,224 \
# --input_dtype=float32 \
# --class_dim=1000 \
# --lr_strategy=cosine_decay \
# --lr=0.1 \
......
......@@ -72,7 +72,7 @@ def check_gpu(use_gpu):
"\t2. Set use_gpu as false in config file to run " \
"model on CPU"
try:
try:
if use_gpu and not fluid.is_compiled_with_cuda():
logger.error(err)
sys.exit(1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册