update model initialization design, supports custom data/num_classes (#44)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Ayush Chaurasia 2022-11-15 20:06:29 +05:30 committed by GitHub
parent 1f3aad86c1
commit 832ea56eb4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 67 additions and 44 deletions

View file

@ -140,8 +140,3 @@ def download(url, dir=Path.cwd(), unzip=True, delete=True, curl=False, threads=1
else:
for u in [url] if isinstance(url, (str, Path)) else url:
download_one(u, dir)
def get_model(model: str):
# check for local weights
pass