v0.1.8

README.md

@@ -38,25 +38,25 @@ pip install .
 python examples/test_env.py
 
 ### cv.py
-pip install "torchvision>=0.13.*"
+pip install "torchvision>=0.13"
 python examples/cv.py
 # Using DP (not recommended, please use DDP)
 python examples/cv.py  # setting device_ids=[0, 1]
 
 ### nlp.py
-pip install "transformers>=4.25.*" "datasets>=2.7.*"
+pip install "transformers>=4.25" "datasets>=2.7"
 python examples/nlp.py
 
 ### dqn.py
 pip install "gym>=0.26.2" "pygame>=2.1.2"
 python examples/dqn.py
 
 ### gan.py
-pip install "torchvision>=0.13.*"
+pip install "torchvision>=0.13"
 python examples/gan.py
 
 ### cl.py  (contrastive_learning)
-pip install "torchvision>=0.13.*" "scikit-learn>=1.2.*"
+pip install "torchvision>=0.13" "scikit-learn>=1.2"
 python examples/cl.py
 
 ### cl_ddp.py
@@ -70,15 +70,15 @@ python examples/gnn2.py
 python examples/gnn3.py
 
 ### ae.py
-pip install "torchvision>=0.13.*" "scikit-learn>=1.2.*"
+pip install "torchvision>=0.13" "scikit-learn>=1.2"
 python examples/ae.py
 
 ### vae.py
-pip install "torchvision>=0.13.*"
+pip install "torchvision>=0.13"
 python examples/vae.py
 
 ### meta_learning.py
-pip install "torchvision>=0.13.*"
+pip install "torchvision>=0.13"
 python examples/meta_learning.py
 
 

requirements.txt

@@ -1,7 +1,7 @@
-torch>=1.12.*
-torchmetrics>=0.11.*
-tensorboard>=2.11.*
-numpy>=1.23.*
+torch>=1.12
+torchmetrics>=0.11
+tensorboard>=2.11
+numpy>=1.23
 PyYAML>=6.0
-tqdm>=4.63.*
-matplotlib>=3.6.*
+tqdm>=4.63
+matplotlib>=3.6