Skip to content

Commit

Permalink
fix bug
Browse files Browse the repository at this point in the history
Signed-off-by: Mengni Wang <[email protected]>
  • Loading branch information
mengniwang95 committed Jun 20, 2024
1 parent 1f6bcee commit 7cc43a9
Showing 1 changed file with 3 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
import os
import sys
import urllib
from urllib

MODEL_URL = "https://github.com/onnx/models/raw/main/validated/vision/classification/resnet/model/resnet50-v1-12.onnx"
MAX_TIMES_RETRY_DOWNLOAD = 5
Expand All @@ -16,8 +15,8 @@ def parse_arguments():


def progressbar(cur, total=100):
percent = '{:.2%}'.format(cur / total)
sys.stdout.write("\r[%-100s] %s" % ('#' * int(cur), percent))
percent = "{:.2%}".format(cur / total)
sys.stdout.write("\r[%-100s] %s" % ("#" * int(cur), percent))
sys.stdout.flush()


Expand Down Expand Up @@ -55,4 +54,4 @@ def prepare_model(input_model, output_model):

if __name__ == "__main__":
args = parse_arguments()
prepare_model(args.input_model, args.output_model)
prepare_model(args.input_model, args.output_model)

0 comments on commit 7cc43a9

Please sign in to comment.