FredZhang7
/

efficientnetv25_rw_s

Image Classification

efficientnetv25

Model card Files Files and versions

FredZhang7 commited on Oct 6, 2023

Commit

0a04b8c

·

1 Parent(s): cac2154

fix bugs

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -43,8 +43,8 @@ print_layer_stats = True    # prints the statistics for each layer of the model
 verbose = True              # prints additional info about the MAC calculation
 # Download the model. Skip this step if already downloaded
-base_model = "efficientnetv2.5_base_in1k.pth"
-url = f"https://huggingface.co/FredZhang7/efficientnetv2.5_rw_s/resolve/main/{model_name}.pth"
 file_name = f"./{base_model}.pth"
 urllib.request.urlretrieve(url, file_name)

 verbose = True              # prints additional info about the MAC calculation
 # Download the model. Skip this step if already downloaded
+base_model = "efficientnetv2.5_base_in1k"
+url = f"https://huggingface.co/FredZhang7/efficientnetv2.5_rw_s/resolve/main/{base_model}.pth"
 file_name = f"./{base_model}.pth"
 urllib.request.urlretrieve(url, file_name)