Update finetune detail, rerun model doc gen.

pull/502/head
Ross Wightman 4 years ago
parent e6f5617dcc
commit 74fcd4c50e

@ -56,7 +56,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('{{ model_name }}', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('{{ model_name }}', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('adv_inception_v3', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('adv_inception_v3', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_efficientnet_b0_ap', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_efficientnet_b0_ap', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('resnetv2_101x1_bitm', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('resnetv2_101x1_bitm', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('cspdarknet53', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('cspdarknet53', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('cspresnet50', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('cspresnet50', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('cspresnext50', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('cspresnext50', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('densenet121', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('densenet121', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('dla102', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('dla102', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('dpn107', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('dpn107', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('ecaresnet101d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('ecaresnet101d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -66,7 +66,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('efficientnet_b1_pruned', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('efficientnet_b1_pruned', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('efficientnet_b0', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('efficientnet_b0', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('ens_adv_inception_resnet_v2', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('ens_adv_inception_resnet_v2', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('ese_vovnet19b_dw', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('ese_vovnet19b_dw', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('fbnetc_100', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('fbnetc_100', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('gluon_inception_v3', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('gluon_inception_v3', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('gluon_resnet101_v1b', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('gluon_resnet101_v1b', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('gluon_resnext101_32x4d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('gluon_resnext101_32x4d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('gluon_senet154', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('gluon_senet154', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('gluon_seresnext101_32x4d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('gluon_seresnext101_32x4d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('gluon_xception65', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('gluon_xception65', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('hrnet_w18', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('hrnet_w18', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('ig_resnext101_32x16d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('ig_resnext101_32x16d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('inception_resnet_v2', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('inception_resnet_v2', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('inception_v3', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('inception_v3', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -59,7 +59,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('inception_v4', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('inception_v4', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('legacy_seresnet101', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('legacy_seresnet101', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('legacy_seresnext101_32x4d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('legacy_seresnext101_32x4d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('legacy_senet154', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('legacy_senet154', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('mixnet_l', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('mixnet_l', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('mnasnet_100', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('mnasnet_100', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('mobilenetv2_100', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('mobilenetv2_100', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('mobilenetv3_large_100', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('mobilenetv3_large_100', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('nasnetalarge', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('nasnetalarge', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -69,7 +69,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_efficientnet_b0_ns', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_efficientnet_b0_ns', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('pnasnet5large', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('pnasnet5large', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('regnetx_002', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('regnetx_002', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -66,7 +66,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('regnety_002', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('regnety_002', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('res2net101_26w_4s', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('res2net101_26w_4s', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('res2next50', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('res2next50', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('resnest101e', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('resnest101e', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('resnet101d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('resnet101d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('resnet18', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('resnet18', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('resnext101_32x8d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('resnext101_32x8d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('rexnet_100', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('rexnet_100', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('seresnet152d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('seresnet152d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('selecsls42b', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('selecsls42b', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('seresnext26d_32x4d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('seresnext26d_32x4d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('skresnet18', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('skresnet18', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('skresnext50_32x4d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('skresnext50_32x4d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('spnasnet_100', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('spnasnet_100', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('ssl_resnet18', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('ssl_resnet18', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('ssl_resnext101_32x16d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('ssl_resnext101_32x16d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('swsl_resnet18', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('swsl_resnet18', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -64,7 +64,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('swsl_resnext101_32x16d', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('swsl_resnext101_32x16d', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -68,7 +68,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_efficientnet_cc_b0_4e', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_efficientnet_cc_b0_4e', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -68,7 +68,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_efficientnet_lite0', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_efficientnet_lite0', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -66,7 +66,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_efficientnet_b0', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_efficientnet_b0', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_inception_v3', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_inception_v3', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_mixnet_l', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_mixnet_l', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tf_mobilenetv3_large_075', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tf_mobilenetv3_large_075', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('tresnet_l', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('tresnet_l', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('vit_base_patch16_224', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('vit_base_patch16_224', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -60,7 +60,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('wide_resnet101_2', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('wide_resnet101_2', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

@ -62,7 +62,7 @@ To extract image features with this model, follow the [timm feature extraction e
## How do I finetune this model? ## How do I finetune this model?
You can finetune any of the pre-trained models just by changing the classifier (the last layer). You can finetune any of the pre-trained models just by changing the classifier (the last layer).
```python ```python
model = timm.create_model('xception', pretrained=True).reset_classifier(NUM_FINETUNE_CLASSES) model = timm.create_model('xception', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)
``` ```
To finetune on your own dataset, you have to write a training loop or adapt [timm's training To finetune on your own dataset, you have to write a training loop or adapt [timm's training
script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset. script](https://github.com/rwightman/pytorch-image-models/blob/master/train.py) to use your dataset.

Loading…
Cancel
Save