!16281 optimized the network of deepfm_test

From: @shuzigood
Reviewed-by: @ouwenchang,@wuxuejian
Signed-off-by: @wuxuejian
This commit is contained in:
mindspore-ci-bot 2021-05-15 16:02:22 +08:00 committed by Gitee
commit a6ae57c9cf
2 changed files with 2 additions and 6 deletions

View File

@ -74,7 +74,7 @@ def convert(weights_file, output_file):
if __name__ == "__main__":
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("--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()
convert(args_opt.input_file, args_opt.output_file)

View File

@ -210,11 +210,7 @@ if __name__ == "__main__":
args, _ = parser.parse_known_args()
base_path = args.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))
print("********tar end***********")
data_stats = DataStatsDict()