diff --git a/__pycache__/defaults.cpython-38.pyc b/__pycache__/defaults.cpython-38.pyc
deleted file mode 100644
index c04a071..0000000
Binary files a/__pycache__/defaults.cpython-38.pyc and /dev/null differ
diff --git a/__pycache__/engine.cpython-38.pyc b/__pycache__/engine.cpython-38.pyc
deleted file mode 100644
index e05be43..0000000
Binary files a/__pycache__/engine.cpython-38.pyc and /dev/null differ
diff --git a/__pycache__/eval_func.cpython-38.pyc b/__pycache__/eval_func.cpython-38.pyc
deleted file mode 100644
index 89b6a26..0000000
Binary files a/__pycache__/eval_func.cpython-38.pyc and /dev/null differ
diff --git a/data/eval_cache/eval_cache.pth b/data/eval_cache/eval_cache.pth
deleted file mode 100644
index 1cee262..0000000
Binary files a/data/eval_cache/eval_cache.pth and /dev/null differ
diff --git a/datasets/__pycache__/__init__.cpython-38.pyc b/datasets/__pycache__/__init__.cpython-38.pyc
deleted file mode 100644
index 1b2abd7..0000000
Binary files a/datasets/__pycache__/__init__.cpython-38.pyc and /dev/null differ
diff --git a/datasets/__pycache__/base.cpython-38.pyc b/datasets/__pycache__/base.cpython-38.pyc
deleted file mode 100644
index c5020d4..0000000
Binary files a/datasets/__pycache__/base.cpython-38.pyc and /dev/null differ
diff --git a/datasets/__pycache__/build.cpython-38.pyc b/datasets/__pycache__/build.cpython-38.pyc
deleted file mode 100644
index 440f1f0..0000000
Binary files a/datasets/__pycache__/build.cpython-38.pyc and /dev/null differ
diff --git a/datasets/__pycache__/cuhk_sysu.cpython-38.pyc b/datasets/__pycache__/cuhk_sysu.cpython-38.pyc
deleted file mode 100644
index 201846d..0000000
Binary files a/datasets/__pycache__/cuhk_sysu.cpython-38.pyc and /dev/null differ
diff --git a/datasets/__pycache__/prw.cpython-38.pyc b/datasets/__pycache__/prw.cpython-38.pyc
deleted file mode 100644
index c8c7b3c..0000000
Binary files a/datasets/__pycache__/prw.cpython-38.pyc and /dev/null differ
diff --git a/loss/__pycache__/oim.cpython-38.pyc b/loss/__pycache__/oim.cpython-38.pyc
deleted file mode 100644
index a9d1b9b..0000000
Binary files a/loss/__pycache__/oim.cpython-38.pyc and /dev/null differ
diff --git a/loss/__pycache__/softmax_loss.cpython-38.pyc b/loss/__pycache__/softmax_loss.cpython-38.pyc
deleted file mode 100644
index 51b1a94..0000000
Binary files a/loss/__pycache__/softmax_loss.cpython-38.pyc and /dev/null differ
diff --git a/.gitignore b/main/.gitignore
similarity index 100%
rename from .gitignore
rename to main/.gitignore
diff --git a/COAT_pt171.yml b/main/COAT_pt171.yml
similarity index 100%
rename from COAT_pt171.yml
rename to main/COAT_pt171.yml
diff --git a/README.md b/main/README.md
similarity index 100%
rename from README.md
rename to main/README.md
diff --git a/configs/cuhk_sysu-local.yaml b/main/configs/cuhk_sysu-local.yaml
similarity index 100%
rename from configs/cuhk_sysu-local.yaml
rename to main/configs/cuhk_sysu-local.yaml
diff --git a/configs/cuhk_sysu.yaml b/main/configs/cuhk_sysu.yaml
similarity index 100%
rename from configs/cuhk_sysu.yaml
rename to main/configs/cuhk_sysu.yaml
diff --git a/configs/prw-local.yaml b/main/configs/prw-local.yaml
similarity index 100%
rename from configs/prw-local.yaml
rename to main/configs/prw-local.yaml
diff --git a/configs/prw.yaml b/main/configs/prw.yaml
similarity index 100%
rename from configs/prw.yaml
rename to main/configs/prw.yaml
diff --git a/datasets/__init__.py b/main/datasets/__init__.py
similarity index 100%
rename from datasets/__init__.py
rename to main/datasets/__init__.py
diff --git a/datasets/base.py b/main/datasets/base.py
similarity index 100%
rename from datasets/base.py
rename to main/datasets/base.py
diff --git a/datasets/build.py b/main/datasets/build.py
similarity index 100%
rename from datasets/build.py
rename to main/datasets/build.py
diff --git a/datasets/cuhk_sysu.py b/main/datasets/cuhk_sysu.py
similarity index 100%
rename from datasets/cuhk_sysu.py
rename to main/datasets/cuhk_sysu.py
diff --git a/datasets/prw.py b/main/datasets/prw.py
similarity index 100%
rename from datasets/prw.py
rename to main/datasets/prw.py
diff --git a/defaults.py b/main/defaults.py
similarity index 100%
rename from defaults.py
rename to main/defaults.py
diff --git a/doc/framework.png b/main/doc/framework.png
similarity index 100%
rename from doc/framework.png
rename to main/doc/framework.png
diff --git a/engine.py b/main/engine.py
similarity index 100%
rename from engine.py
rename to main/engine.py
diff --git a/eval_func.py b/main/eval_func.py
similarity index 100%
rename from eval_func.py
rename to main/eval_func.py
diff --git a/loss/oim.py b/main/loss/oim.py
similarity index 100%
rename from loss/oim.py
rename to main/loss/oim.py
diff --git a/loss/softmax_loss.py b/main/loss/softmax_loss.py
similarity index 100%
rename from loss/softmax_loss.py
rename to main/loss/softmax_loss.py
diff --git a/models/coat.py b/main/models/coat.py
similarity index 100%
rename from models/coat.py
rename to main/models/coat.py
diff --git a/models/resnet.py b/main/models/resnet.py
similarity index 100%
rename from models/resnet.py
rename to main/models/resnet.py
diff --git a/models/transformer.py b/main/models/transformer.py
similarity index 100%
rename from models/transformer.py
rename to main/models/transformer.py
diff --git a/train.py b/main/train.py
similarity index 100%
rename from train.py
rename to main/train.py
diff --git a/utils/km.py b/main/utils/km.py
similarity index 100%
rename from utils/km.py
rename to main/utils/km.py
diff --git a/utils/mask.py b/main/utils/mask.py
similarity index 100%
rename from utils/mask.py
rename to main/utils/mask.py
diff --git a/utils/transforms.py b/main/utils/transforms.py
similarity index 100%
rename from utils/transforms.py
rename to main/utils/transforms.py
diff --git a/utils/utils.py b/main/utils/utils.py
similarity index 100%
rename from utils/utils.py
rename to main/utils/utils.py
diff --git a/vis/results.json b/main/vis/results.json
similarity index 100%
rename from vis/results.json
rename to main/vis/results.json
diff --git a/models/__pycache__/coat.cpython-38.pyc b/models/__pycache__/coat.cpython-38.pyc
deleted file mode 100644
index 34bb06a..0000000
Binary files a/models/__pycache__/coat.cpython-38.pyc and /dev/null differ
diff --git a/models/__pycache__/resnet.cpython-38.pyc b/models/__pycache__/resnet.cpython-38.pyc
deleted file mode 100644
index 98caa71..0000000
Binary files a/models/__pycache__/resnet.cpython-38.pyc and /dev/null differ
diff --git a/models/__pycache__/transformer.cpython-38.pyc b/models/__pycache__/transformer.cpython-38.pyc
deleted file mode 100644
index 7aa588c..0000000
Binary files a/models/__pycache__/transformer.cpython-38.pyc and /dev/null differ
diff --git a/utils/__pycache__/km.cpython-38.pyc b/utils/__pycache__/km.cpython-38.pyc
deleted file mode 100644
index bed8001..0000000
Binary files a/utils/__pycache__/km.cpython-38.pyc and /dev/null differ
diff --git a/utils/__pycache__/mask.cpython-38.pyc b/utils/__pycache__/mask.cpython-38.pyc
deleted file mode 100644
index c7aad48..0000000
Binary files a/utils/__pycache__/mask.cpython-38.pyc and /dev/null differ
diff --git a/utils/__pycache__/transforms.cpython-38.pyc b/utils/__pycache__/transforms.cpython-38.pyc
deleted file mode 100644
index 349aebf..0000000
Binary files a/utils/__pycache__/transforms.cpython-38.pyc and /dev/null differ
diff --git a/utils/__pycache__/utils.cpython-38.pyc b/utils/__pycache__/utils.cpython-38.pyc
deleted file mode 100644
index cc00710..0000000
Binary files a/utils/__pycache__/utils.cpython-38.pyc and /dev/null differ