From 942032ea810091aaebe6fcc6946068ed98244459 Mon Sep 17 00:00:00 2001
From: XinyangHan <2544130289@qq.com>
Date: Mon, 25 Dec 2023 18:37:28 +0000
Subject: [PATCH] test

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

diff --git a/README.md b/README.md
index daa2989..b08a895 100644
--- a/README.md
+++ b/README.md
@@ -43,6 +43,8 @@ python ./tools/train_net.py  --config-file ./configs/COCO-PanopticSegmentation/u
 ```
 Note: you need to download the pre-trained [dino backbone](https://drive.google.com/file/d/1UtRUgUQK20KS8MGebCWgLPHxrez7mfV4/view?usp=sharing) and change the path of the corresponding ```yaml``` file.
 
+To train U2Seg model with different clustering number (e.g. 300), you can use `configs/COCO-PanopticSegmentation/u2seg_R50_300.yaml` config file and set the environment variable by `export CLUSTER_NUM=300`.
+
 ### Demonstration
 
 ### Model Zoo