upgrade pytorch-lightning and hdyra

hyperbox/configs/hydra/default.yaml

@@ -11,3 +11,5 @@ job:
   env_set:
     EXAMPLE_VAR: "example_value"
   name: "exp"
+  chdir: True # the output/working directory will be changed to {hydra.job.name}, you can see below URL for more details
+  # https://hydra.cc/docs/next/tutorials/basic/running_your_app/working_directory/#disable-changing-current-working-dir-to-jobs-output-dir

hyperbox/run.py

@@ -1,17 +1,17 @@
-import sys
 import os
+import sys
 
 import dotenv
 import hydra
-from omegaconf import DictConfig
 from ipdb import set_trace
+from omegaconf import DictConfig
 
 # load environment variables from `.env` file if it exists
 # recursively searches for `.env` in all folders starting from work dir
 dotenv.load_dotenv(override=True)
 
 
-@hydra.main(config_path="configs/", config_name="config.yaml")
+@hydra.main(version_base="1.2", config_path="configs/", config_name="config.yaml")
 def main(config: DictConfig):
 
     # Imports should be nested inside @hydra.main to optimize tab completion

requirements.txt

@@ -1,19 +1,19 @@
 # --------- pytorch --------- #
 torch>=1.8.1
-pytorch-lightning>=1.5.8
+pytorch-lightning>=1.6.0
 torchmetrics>=0.3.2
 lightning-bolts>=0.3.3
 
 # --------- hydra --------- #
-hydra-core==1.1.0.rc1
+hydra-core==1.2.0
 hydra-colorlog==1.1.0.dev1
 hydra-optuna-sweeper==1.1.0.dev2
 # hydra-ax-sweeper==1.1.0
 # hydra-ray-launcher==0.1.2
 # hydra-submitit-launcher==1.1.0
 
 # --------- loggers --------- #
-wandb>=0.10.30
+wandb>=0.12.21
 # neptune-client
 # mlflow
 # comet-ml

setup.py

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