logo
Browse Source

add training requirements readme

main
ChengZi 2 years ago
parent
commit
5ad5ad40b0
  1. 6
      README.md

6
README.md

@ -114,7 +114,13 @@ op.save_model('onnx', 'test.onnx')
## 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.
```python
! python -m pip install tqdm augly timm==0.4.12 pytorch-metric-learning==0.9.99
```
### Get start to fine-tune
Just call method op.train() and pass in your args.

Loading…
Cancel
Save