forked from mindspore-Ecosystem/mindspore
optimized the network of deepfm_test
This commit is contained in:
parent
eb80f555b7
commit
e792473ea4
|
@ -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)
|
||||
|
|
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue