Update README.md
Browse files
README.md
CHANGED
|
@@ -24,6 +24,24 @@ ImageNet-1K results
|
|
| 24 |
| -- | -- |
|
| 25 |
| 80.9695 | 95.1044 |
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
training command
|
| 28 |
|
| 29 |
```bash
|
|
|
|
| 24 |
| -- | -- |
|
| 25 |
| 80.9695 | 95.1044 |
|
| 26 |
|
| 27 |
+
config
|
| 28 |
+
|
| 29 |
+
```python
|
| 30 |
+
@register_model
|
| 31 |
+
def resnext101d_32x4d(pretrained: bool = False, **kwargs) -> ResNet:
|
| 32 |
+
"""Constructs a ResNeXt101d 32x4d model."""
|
| 33 |
+
model_args = dict(
|
| 34 |
+
block=Bottleneck,
|
| 35 |
+
layers=(3, 4, 23, 3),
|
| 36 |
+
cardinality=32,
|
| 37 |
+
base_width=4,
|
| 38 |
+
stem_width=32,
|
| 39 |
+
stem_type="deep",
|
| 40 |
+
avg_down=True,
|
| 41 |
+
)
|
| 42 |
+
return _create_resnet("resnext101d_32x4d", pretrained, **dict(model_args, **kwargs))
|
| 43 |
+
```
|
| 44 |
+
|
| 45 |
training command
|
| 46 |
|
| 47 |
```bash
|