Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MobileNetV3 Architecture in TorchVision #3182

Merged
merged 26 commits into from
Jan 5, 2021
Merged

Add MobileNetV3 Architecture in TorchVision #3182

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
22b9b12
partial implementation network architecture
datumbox Dec 15, 2020
0cff6fb
Simplify implementation and adding blocks.
datumbox Dec 16, 2020
fd54fdf
Refactoring the code to make it more readable.
datumbox Dec 16, 2020
834b185
Adding first conv layers.
datumbox Dec 16, 2020
1edd16b
Moving mobilenet.py to mobilenetv2.py
datumbox Dec 16, 2020
2f52f0d
Adding mobilenet.py for BC.
datumbox Dec 16, 2020
bb2ec9e
Extending ConvBNReLU for reuse.
datumbox Dec 16, 2020
e95ee5c
Moving mobilenet.py to mobilenetv2.py
datumbox Dec 16, 2020
2ebe8ba
Adding mobilenet.py for BC.
datumbox Dec 16, 2020
0c31a33
Extending ConvBNReLU for reuse.
datumbox Dec 16, 2020
db7522b
Reduce import scope on mobilenet to only the public and versioned cla…
datumbox Dec 16, 2020
fdbcec7
Merge branch 'refactoring/mobilenetv2_bc_move' into models/mobilenetv3
datumbox Dec 16, 2020
16f55f5
Further simplify by reusing MobileNetv2 methods.
datumbox Dec 16, 2020
8162fa4
Adding the remaining implementation of mobilenetv3.
datumbox Dec 16, 2020
8615585
Adding tests, docs and init methods.
datumbox Dec 16, 2020
8664fde
Refactoring and fixing formatter.
datumbox Dec 16, 2020
cfa20b7
Fixing type issues.
datumbox Dec 16, 2020
c189ae1
Using build-in Hardsigmoid and Hardswish.
datumbox Dec 16, 2020
5030435
Merge branch 'master' into models/mobilenetv3
datumbox Dec 17, 2020
9a758a8
Code review nits.
datumbox Dec 17, 2020
25f8b26
Putting inplace on Dropout.
datumbox Dec 17, 2020
5198385
Adding rmsprop support on the train.py
datumbox Jan 1, 2021
e4d130f
Adding auto-augment and random-erase in the training scripts.
datumbox Jan 1, 2021
5d0a664
Merge branch 'master' into models/mobilenetv3
datumbox Jan 3, 2021
c0a13a2
Adding support for reduced tail on MobileNetV3.
datumbox Jan 5, 2021
2414d2d
Tagging blocks with comments.
datumbox Jan 5, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Adding mobilenet.py for BC.
  • Loading branch information
@datumbox
datumbox committed Dec 16, 2020
commit 2f52f0dd460a77a2336dc20c1d02083a9069aa96
1 change: 1 addition & 0 deletions torchvision/models/mobilenet.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from .mobilenetv2 import *
1 change: 1 addition & 0 deletions torchvision/models/quantization/mobilenet.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from .mobilenetv2 import *
2 changes: 1 addition & 1 deletion torchvision/models/quantization/mobilenetv2.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
from torch import nn
from torchvision.models.utils import load_state_dict_from_url
from torchvision.models.mobilenet import InvertedResidual, ConvBNReLU, MobileNetV2, model_urls
from torchvision.models.mobilenetv2 import InvertedResidual, ConvBNReLU, MobileNetV2, model_urls
from torch.quantization import QuantStub, DeQuantStub, fuse_modules
from .utils import _replace_relu, quantize_model

Expand Down