Skip to content

Commit

Permalink
update steps for 1D PDE data generation
Browse files Browse the repository at this point in the history
  • Loading branch information
kmario23 authored Oct 2, 2024
1 parent 219e544 commit b11eab3
Showing 1 changed file with 31 additions and 3 deletions.
34 changes: 31 additions & 3 deletions pdebench/data_gen/data_gen_NLE/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,18 @@ CUDA_VISIBLE_DEVICES='2,3' python3 advection_multi_solution_Hydra.py +multi=beta
# Or generate all files
bash run_trainset.sh
```

- Update `data_gen/data_gen_NLE/config/config.yaml` to:

```yaml
type: 'advection' # 'advection'/'ReacDiff'/'burgers'/'CFD'
dim: 1
savedir: './save/advection'
```
```
# serialize to hdf5 by transforming npy file
# before executing Data_Merge.py, change "type" and "savedir" values in PDEBench/pdebench/data_gen/data_gen_NLE/config/config.yaml
cd ..
python Data_Merge.py
```
Expand All @@ -56,9 +65,19 @@ CUDA_VISIBLE_DEVICES='0,2' python3 burgers_multi_solution_Hydra.py +multi=1e-1.y
# Or generate all files
bash run_trainset.sh
```

- Update `data_gen/data_gen_NLE/config/config.yaml` to:

```yaml
type: 'burgers' # 'advection'/'ReacDiff'/'burgers'/'CFD'
dim: 1
savedir: './save/burgers'
```
```
# serialize to hdf5 by transforming npy file
# before executing Data_Merge.py, change "type" and "savedir" values in PDEBench/pdebench/data_gen/data_gen_NLE/config/config.yaml
cd ..
python Data_Merge.py
```
Expand All @@ -76,9 +95,18 @@ CUDA_VISIBLE_DEVICES='0,1' python3 reaction_diffusion_multi_solution_Hydra.py +m
# Or generate all files
bash run_trainset.sh
```

- Update `data_gen/data_gen_NLE/config/config.yaml` to:

```yaml
type: 'ReacDiff' # 'advection'/'ReacDiff'/'burgers'/'CFD'
dim: 1
savedir: './save/ReacDiff'
```
```
# serialize to hdf5 by transforming npy file
# before executing Data_Merge.py, change "type" and "savedir" values in PDEBench/pdebench/data_gen/data_gen_NLE/config/config.yaml
cd ..
python Data_Merge.py
```

0 comments on commit b11eab3

Please sign in to comment.