From 5ad5ad40b0c7ed3eb2e752c50ae311336629ce1a Mon Sep 17 00:00:00 2001
From: ChengZi <chen.zhang@zilliz.com>
Date: Thu, 5 Jan 2023 10:29:46 +0800
Subject: [PATCH] add training requirements readme

---
 README.md | 6 ++++++
 1 file changed, 6 insertions(+)

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.