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

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()