v0.1.5

README.md

@@ -1,4 +1,9 @@
 # Mini-Lightning
+[![Python Version](https://img.shields.io/pypi/pyversions/mini-lightning)](https://pypi.org/project/mini-lightning/)
+[![Pytorch Version](https://img.shields.io/badge/pytorch-1.12-blue.svg)](https://pypi.org/project/mini-lightning/)
+[![PyPI Status](https://badge.fury.io/py/mini-lightning.svg)](https://badge.fury.io/py/mini-lightning)
+[![license](https://img.shields.io/badge/License-MIT-blue.svg)](https://github.com/ustcml/mini-lightning/blob/main/LICENSE)
+[![Downloads](https://pepy.tech/badge/mini-lightning)](https://pepy.tech/project/mini-lightning)
 
 
 ## Introduction
@@ -15,7 +20,7 @@
 3. Install mini-lightning
 ```bash
 # from pypi
-pip install mini-lightning==0.1.4.1
+pip install mini-lightning==0.1.5
 
 # Or download the files from the repository to local,
 # and go to the folder where setup.py is located, and run the following command
@@ -39,7 +44,7 @@ python examples/cv.py
 python examples/cv.py  # setting device_ids=[0, 1]
 
 ### nlp.py
-pip install "transformers>=4.22.*" "datasets>=2.5.*"
+pip install "transformers>=4.23.*" "datasets>=2.6.*"
 python examples/nlp.py
 
 ### dqn.py

mini_lightning/_mini_lightning.py

@@ -724,7 +724,7 @@ def _val_test(
         lmodel.model = de_parallel(lmodel.model)
         metrics_r: Dict[str, bool] = {k: m._to_sync for k, m in lmodel.metrics.items()}
         for m in lmodel.metrics.values():
-            # torchmetrics(>=0.9.3, <=0.10.0) private variable. I don't know whether it will be changed later.
+            # torchmetrics(>=0.9.3, <=0.10.1已测试) private variable. I don't know whether it will be changed later.
             #   You can raise issue if finding error.
             # default: sync_on_compute = True
             m._to_sync = False

requirements.txt

@@ -1,6 +1,6 @@
 torch>=1.12.*
-torchmetrics>=0.10.0
-tensorboard
-numpy
-PyYAML
-tqdm
+torchmetrics>=0.10.*
+tensorboard>=2.10.*
+numpy>=1.23.*
+PyYAML>=6.0
+tqdm>=4.64.*