Skip to content

Commit 97fc329

Browse files
committed
Make Inception models work
1. Transition to `Inceptionv3` instead of `Inception3` for standardisation of names 2. Formatting
1 parent f61d362 commit 97fc329

File tree

6 files changed

+136
-125
lines changed

6 files changed

+136
-125
lines changed

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
name = "Metalhead"
22
uuid = "dbeba491-748d-5e0e-a39e-b530a07fa0cc"
3-
version = "0.7.2"
3+
version = "0.7.3-DEV"
44

55
[deps]
66
Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
| [VGG](https://arxiv.org/abs/1409.1556) | [`VGG`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.VGG.html) | N |
2020
| [ResNet](https://arxiv.org/abs/1512.03385) | [`ResNet`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.ResNet.html) | N |
2121
| [GoogLeNet](https://arxiv.org/abs/1409.4842) | [`GoogLeNet`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.GoogLeNet.html) | N |
22-
| [Inception-v3](https://arxiv.org/abs/1512.00567) | [`Inception3`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.Inception3.html) | N |
22+
| [Inception-v3](https://arxiv.org/abs/1512.00567) | [`Inceptionv3`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.Inceptionv3.html) | N |
2323
| [SqueezeNet](https://arxiv.org/abs/1602.07360) | [`SqueezeNet`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.SqueezeNet.html) | N |
2424
| [DenseNet](https://arxiv.org/abs/1608.06993) | [`DenseNet`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.DenseNet.html) | N |
2525
| [ResNeXt](https://arxiv.org/abs/1611.05431) | [`ResNeXt`](https://fluxml.ai/Metalhead.jl/dev/docstrings/Metalhead.ResNeXt.html) | N |

src/Metalhead.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,8 @@ export AlexNet,
4141
VGG, VGG11, VGG13, VGG16, VGG19,
4242
GoogLeNet,
4343
ResNet, ResNet18, ResNet34, ResNet50, ResNet101, ResNet152,
44-
Inception3, Inception4, SqueezeNet,
44+
Inception3, Inceptionv3, Inceptionv4, InceptionResNetv2,
45+
SqueezeNet,
4546
ResNeXt,
4647
DenseNet, DenseNet121, DenseNet161, DenseNet169, DenseNet201,
4748
MobileNetv1, MobileNetv2, MobileNetv3,
@@ -50,7 +51,7 @@ export AlexNet,
5051
ConvMixer, ConvNeXt
5152

5253
# use Flux._big_show to pretty print large models
53-
for T in (:AlexNet, :VGG, :GoogLeNet, :ResNet, :ResNeXt, :Inception3, :Inception4,
54+
for T in (:AlexNet, :VGG, :GoogLeNet, :ResNet, :ResNeXt, :Inceptionv3,
5455
:SqueezeNet, :DenseNet, :MobileNetv1, :MobileNetv2, :MobileNetv3,
5556
:MLPMixer, :ResMLP, :gMLP, :ViT, :ConvMixer, :ConvNeXt)
5657
@eval Base.show(io::IO, ::MIME"text/plain", model::$T) = _maybe_big_show(io, model)

0 commit comments

Comments
 (0)