Col_lin
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bnfold update

This commit is contained in:
zhang__sss 2021-05-24 17:35:34 +08:00
parent 71b522ec09
commit e6a4b0ed8a
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ops/_op_impl/_custom_op/batchnorm_fold.py
View File

@ -28,11 +28,11 @@ batch_norm_op_info = TBERegOp("BatchNormFoldD") \
.compute_cost(10) \
.kernel_name("batchnorm_fold") \
.partial_flag(True) \
.attr("momentum", "optional", "float", "all") \
.attr("epsilon", "optional", "float", "all") \
.attr("is_training", "optional", "bool", "all") \
.attr("freeze_bn", "optional", "int", "all") \
.attr("format", "optional", "str", "all") \
.attr("momentum", "optional", "float", "all", "0.9") \
.attr("epsilon", "optional", "float", "all", "0.00001") \
.attr("is_training", "optional", "bool", "all", "true") \
.attr("freeze_bn", "optional", "int", "all", "0") \
.attr("format", "optional", "str", "all", "NCHW") \
.input(0, "x", False, "required", "all") \
.input(1, "x_sum", False, "required", "all") \
.input(2, "x_square_sum", False, "required", "all") \

2
model_zoo/official/cv/yolov3_darknet53_quant/eval.py
View File

@ -37,7 +37,7 @@ from src.yolo_dataset import create_yolo_dataset
from src.config import ConfigYOLOV3DarkNet53
devid = int(os.getenv('DEVICE_ID'))
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=True, device_id=devid)
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False, device_id=devid)
class Redirct:

2
model_zoo/official/cv/yolov3_darknet53_quant/train.py
View File

@ -44,7 +44,7 @@ set_seed(1)
devid = int(os.getenv('DEVICE_ID'))
context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True,
device_target="Ascend", save_graphs=True, device_id=devid)
device_target="Ascend", save_graphs=False, device_id=devid)
def parse_args():