diff --git a/README.md b/README.md index 825089c..6fdd06f 100644 --- a/README.md +++ b/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.