update before release

README.md

@@ -20,13 +20,13 @@ conda create -n mbls python=3.8.10
 ```
 
 It's recommended to install [torch and torchvision](https://pytorch.org/) tailored to your environment in advance.
-The torch versions I have tested are 1.10.0+cu111 and 1.7.1+cu110.
+The torch versions I have tested are **1.10.0+cu111** and 1.7.1+cu110.
 
 ```
 pip install -e .
 ```
 
-The required libraies are included in [steup.py](setup.py).
+The required libraies are already included in [steup.py](setup.py).
 
 ## Data preparation
 

docs/TEST.md

@@ -14,6 +14,12 @@ python tools/test_net.py \
     test.checkpoint=[The Filename Of The Checkpoint]
 ```
 
+### Checkpoints links
+
+[CE]()
+[LS]()
+[FL]()
+[MbLS(Ours)]()
 
 ### Running Examples
 
@@ -69,8 +75,34 @@ python tools/test_net.py \
 +---------+---------+---------+---------+---------+
 ```
 
+### FL
+```python
+python tools/test_net.py \
+    data=tiny_imagenet \
+    data.data_root=/home/bliu/work/Data/tiny-imagenet-200 \
+    model=resnet50_tiny \
+    model.num_classes=200 \
+    hydra.run.dir=outputs/best_models/tiny_resnet50 \
+    test.checkpoint=resnet50_tiny-fl-best.pth
+```
+
+```
+[2022-04-30 18:37:37,130 INFO][tester.py:123 - log_eval_epoch_info] -
++---------+---------+---------+---------+
+| samples | acc     | acc_5   | macc    |
++---------+---------+---------+---------+
+| 10000   | 0.63090 | 0.85600 | 0.63090 |
++---------+---------+---------+---------+
+[2022-04-30 18:37:37,130 INFO][tester.py:124 - log_eval_epoch_info] -
++---------+---------+---------+---------+---------+
+| samples | nll     | ece     | aece    | cece    |
++---------+---------+---------+---------+---------+
+| 10000   | 1.44640 | 0.02958 | 0.03116 | 0.00139 |
++---------+---------+---------+---------+---------+
+```
+
 
-#### MbLS
+#### MbLS (Ours)
 ```python
 python tools/test_net.py \
     data=tiny_imagenet \

setup.py

@@ -2,8 +2,8 @@
 
 setup(
     name="calibrate",
-    version="1.0",
-    author="",
+    version="0.2",
+    author="Bingyuan Liu",
     description="For awesome calibration research",
     packages=find_packages(),
     python_requries=">=3.8",