Skip to content

Commit 44de174

Browse files
authored
Apply suggestions from code review
1 parent cabe8aa commit 44de174

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/Metalhead.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ export AlexNet,
3838
# use Flux._big_show to pretty print large models
3939
for T in (:AlexNet, :VGG, :ResNet, :GoogLeNet, :Inception3, :SqueezeNet, :DenseNet, :ResNeXt,
4040
:MobileNetv2, :MobileNetv3, :MLPMixer)
41-
@eval Base.show(io::IO, ::MIME"text/plain", model::$T) = _maybe_big_show(io, model)
41+
@eval Base.show(io::IO, ::MIME"text/plain", model::$T) = _maybe_big_show(io, model)
4242
end
4343

4444
end # module

src/other/mlpmixer.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,4 +84,4 @@ end
8484
(m::MLPMixer)(x) = m.layers(x)
8585

8686
backbone(m::MLPMixer) = m.layers[1]
87-
classifier(m::MLPMixer) = m.layers[2:end]
87+
classifier(m::MLPMixer) = m.layers[2]

0 commit comments

Comments
 (0)