p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

optimized the network of deepfm_test

This commit is contained in:
wsq3 2021-05-12 10:44:26 +08:00
parent eb80f555b7
commit e792473ea4
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/yolov3_darknet53/src/convert_weight.py
View File

@ -74,7 +74,7 @@ def convert(weights_file, output_file):
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser(description="yolov3 weight convert.") parser = argparse.ArgumentParser(description="yolov3 weight convert.")
parser.add_argument("--input_file", type=str, default="./darknet53.conv.74", help="input file path.") parser.add_argument("--input_file", type=str, default="./darknet53.conv.74", help="input file path.")
parser.add_argument("--output_file", type=str, default="./ackbone_darknet53.ckpt", help="output file path.") parser.add_argument("--output_file", type=str, default="./backbone_darknet53.ckpt", help="output file path.")
args_opt = parser.parse_args() args_opt = parser.parse_args()
convert(args_opt.input_file, args_opt.output_file) convert(args_opt.input_file, args_opt.output_file)

6
tests/st/model_zoo_tests/DeepFM/process_data.py
View File

@ -210,11 +210,7 @@ if __name__ == "__main__":
args, _ = parser.parse_known_args() args, _ = parser.parse_known_args()
base_path = args.base_path base_path = args.base_path
data_path = base_path + "" data_path = base_path + ""
# mkdir_path(data_path)
# if not os.path.exists(base_path + "dac.tar.gz"):
# os.system(
# "wget -P {} -c https://s3-eu-west-1.amazonaws.com/kaggle-display-advertising-challenge-dataset/dac.tar.gz --no-check-certificate".format(
# base_path))
os.system("tar -zxvf {}dac.tar.gz".format(data_path)) os.system("tar -zxvf {}dac.tar.gz".format(data_path))
print("********tar end***********") print("********tar end***********")
data_stats = DataStatsDict() data_stats = DataStatsDict()