v1.2.0

marsggbo · Jul 19, 2022 · a52e060 · a52e060
1 parent da828be
commit a52e060
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -33,6 +33,7 @@ pip install hyperbox
 ```
 git clone https://github.com/marsggbo/hyperbox
 cd hyperbox
+python install -r requirements.txt
 python setup.py develop
 ```
 
@@ -42,7 +43,7 @@ python setup.py develop
 <summary><b> Quick Start </b></summary>
 
 ```
-python -m hyperbox.run experimen=example_random trainer.fast_dev_run=True
+python -m hyperbox.run experiment=example_random_nas trainer.fast_dev_run=True
 ```
 
 </details>

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 # --------- pytorch --------- #
 torch>=1.8.1
+torchvision
 pytorch-lightning>=1.6.0
 torchmetrics>=0.3.2
 lightning-bolts>=0.3.3
@@ -26,6 +27,8 @@ isort
 flake8
 
 # --------- others --------- #
+ipdb
+loguru
 jupyterlab
 python-dotenv
 rich

diff --git a/setup.py b/setup.py
@@ -4,11 +4,11 @@
 setup(
     name="hyperbox",  # you should change "src" to your project name
     version="1.2.0",
-    description="Hyperbox: An easy-to-use NAS framework.",
+    description="Hyperbox: An easy-to-use AutoML framework.",
     author="marsggbo",
     url="https://github.com/marsggbo/hyperbox",
     # replace with your own github project link
-    install_requires=["pytorch-lightning>=1.6", "hydra-core>=1.1"],
+    install_requires=["pytorch-lightning>=1.6", "hydra-core>=1.2"],
     packages=find_packages(),
     include_package_data=True,
 )