ChengZi
2 years ago
1 changed files with
6 additions and
0 deletions
-
README.md
|
|
@ -114,7 +114,13 @@ op.save_model('onnx', 'test.onnx') |
|
|
|
|
|
|
|
|
## Fine-tune |
|
|
## Fine-tune |
|
|
|
|
|
|
|
|
|
|
|
### Requirements |
|
|
|
|
|
|
|
|
If you want to fine tune this operator, make sure your timm version is 0.4.12, and the higher version would cause model collapse during training. |
|
|
If you want to fine tune this operator, make sure your timm version is 0.4.12, and the higher version would cause model collapse during training. |
|
|
|
|
|
```python |
|
|
|
|
|
! python -m pip install tqdm augly timm==0.4.12 pytorch-metric-learning==0.9.99 |
|
|
|
|
|
``` |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
### Get start to fine-tune |
|
|
### Get start to fine-tune |
|
|
Just call method op.train() and pass in your args. |
|
|
Just call method op.train() and pass in your args. |
|
|
|
