Fixed code for running

.gitignore

@@ -224,3 +224,9 @@ poetry.toml
 pyrightconfig.json
 
 # End of https://www.toptal.com/developers/gitignore/api/python
+
+
+
+outputs/
+data/
+param_data/

README.md

@@ -35,7 +35,7 @@ git clone https://github.com/NUS-HPC-AI-Lab/Neural-Network-Diffusion.git
 2. Create a new Conda environment and activate it: 
 
 ```
-conda env create -f environment.yaml
+conda env create -f environment.yml
 conda activate pdiff
 ```
 

cifar100_resnet18_k200.sh

@@ -1,2 +1,2 @@
-python task_training # get the model data for training p-diff
-python train_p_diff --task cifar100 --system ae_ddpm 
+python task_training.py # get the model data for training p-diff
+python train_p_diff.py task=cifar100 system=ae_ddpm 

configs/base.yaml

@@ -5,7 +5,7 @@ defaults:
 
 device:
   cuda_visible_devices:  "0,1,2,3,4,5,6"
-  id: 1
+  id: 0
   cuda: cuda:${device.id}
 
 load_system_checkpoint:

core/system/__init__.py

@@ -3,12 +3,12 @@
 from .encoder import *
 from .vae import VAESystem
 from .ae_ddpm import AE_DDPM
-from .explore import Explore
+# from .explore import Explore
 
 systems = {
     'encoder': EncoderSystem,
     'ddpm': DDPM,
     'vae': VAESystem,
     'ae_ddpm': AE_DDPM,
-    'explore': Explore,
+    # 'explore': Explore,
 }