@@ -6,13 +6,13 @@ Creates a SEResNet model with the specified depth.
6
6
7
7
# Arguments
8
8
9
- - `depth`: one of `[18, 34, 50, 101, 152]`. The depth of the ResNet model.
9
+ - `depth`: one of `[18, 34, 50, 101, 152]`. The depth of the SEResNet model.
10
10
- `pretrain`: set to `true` to load the model with pre-trained weights for ImageNet
11
11
- `inchannels`: the number of input channels.
12
12
- `nclasses`: the number of output classes
13
13
14
14
!!! warning
15
-
15
+
16
16
`SEResNet` does not currently support pretrained weights.
17
17
18
18
Advanced users who want more configuration options will be better served by using [`resnet`](@ref).
@@ -49,15 +49,15 @@ Creates a SEResNeXt model with the specified depth, cardinality, and base width.
49
49
50
50
# Arguments
51
51
52
- - `depth`: one of `[50, 101, 152]`. The depth of the ResNet model.
52
+ - `depth`: one of `[50, 101, 152]`. The depth of the SEResNeXt model.
53
53
- `pretrain`: set to `true` to load the model with pre-trained weights for ImageNet
54
54
- `cardinality`: the number of groups to be used in the 3x3 convolution in each block.
55
55
- `base_width`: the number of feature maps in each group.
56
56
- `inchannels`: the number of input channels
57
57
- `nclasses`: the number of output classes
58
58
59
59
!!! warning
60
-
60
+
61
61
`SEResNeXt` does not currently support pretrained weights.
62
62
63
63
Advanced users who want more configuration options will be better served by using [`resnet`](@ref).
0 commit comments