From 91a7cd8683721744dd532a2dd0e78f8a7c792152 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Wed, 15 Feb 2023 21:12:05 +0800 Subject: [PATCH] v0.1.8 --- README.md | 14 +++++++------- requirements.txt | 12 ++++++------ 2 files changed, 13 insertions(+), 13 deletions(-) diff --git a/README.md b/README.md index 2f85f5a..0ee9527 100644 --- a/README.md +++ b/README.md @@ -38,13 +38,13 @@ 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 @@ -52,11 +52,11 @@ 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 diff --git a/requirements.txt b/requirements.txt index 9767f20..aa038bc 100644 --- a/requirements.txt +++ b/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.* \ No newline at end of file +tqdm>=4.63 +matplotlib>=3.6 \ No newline at end of file