From df799c479696d5c71fff1310139af5e79fc97821 Mon Sep 17 00:00:00 2001 From: Martin Garbade Date: Wed, 4 Mar 2020 10:27:00 +0100 Subject: [PATCH 1/2] fixed downloading of model weights --- README.md | 4 +++- download_weight.sh | 7 +++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 3c6acca..11e1f4d 100644 --- a/README.md +++ b/README.md @@ -17,7 +17,9 @@ Python packages: ## Usage ### Download weights - +Download manually from [here](https://github.com/d-li14/mobilenetv3.pytorch/raw/master/pretrained/mobilenetv3-large-657e7b3d.pth) +or use + ./download_weight.sh ### Run the script diff --git a/download_weight.sh b/download_weight.sh index 876bc25..59f7ffd 100755 --- a/download_weight.sh +++ b/download_weight.sh @@ -1,7 +1,6 @@ #!/usr/bin/env bash -git clone git@github.com:d-li14/mobilenetv3.pytorch.git ./tmp - mkdir pretrained -cp ./tmp/pretrained/* ./pretrained/ -rm -rf ./tmp \ No newline at end of file +cd pretrained +wget https://github.com/d-li14/mobilenetv3.pytorch/raw/master/pretrained/mobilenetv3-large-657e7b3d.pth + From e0fd8d78673a6cc8ece9f44ce2514bcb73016d67 Mon Sep 17 00:00:00 2001 From: Martin Garbade Date: Wed, 4 Mar 2020 10:28:42 +0100 Subject: [PATCH 2/2] fixed display of logging output --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index 217112c..94e1e0d 100644 --- a/main.py +++ b/main.py @@ -10,6 +10,7 @@ logger = logging.getLogger() logger.setLevel(logging.INFO) +logger.addHandler(logging.StreamHandler()) def main():