Merge branch 'master'

marsggbo · Jul 29, 2022 · f78a3aa · f78a3aa
2 parents 1eca85a + 884e262
commit f78a3aa
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 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
@@ -36,4 +39,5 @@ seaborn
 pudb
 colorlog
 graphviz
+peewee # for building nasbench-201
 # dvc
diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
     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,
 )