Skip to content

Commit

Permalink
updated requirements
Browse files Browse the repository at this point in the history
  • Loading branch information
Prantl committed Nov 2, 2022
1 parent 3c7b4dd commit 305d41b
Show file tree
Hide file tree
Showing 7 changed files with 6 additions and 8 deletions.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ Please cite [our paper](https://openreview.net/pdf?id=6niwHlzh10U) if you find t

Used environment: python3.7 with CUDA 11.3 and CUDNN 8.0.
- Install libcap-dev: ```sudo apt install libcap-dev```
- Install cmake: ```sudo apt install cmake```
- Update pip: ```pip install --upgrade pip```
- Install requirements: ```pip install -r requirements.txt```
- Tensorpack DataFlow ```pip install --upgrade git+https://github.com/tensorpack/dataflow.git```

Expand Down Expand Up @@ -74,7 +76,7 @@ python run_pipeline.py --cfg_file configs/WBC-SPH.yml \
```
*Note: The argument ```pipeline.data_generator.test.time_end```, ```pipeline.data_generator.valid.time_end```, ```pipeline.data_generator.valid.random_start```, and ```pipeline.test_compute_metric``` are examples how to overwrite corresponding entries in the config file.*

The ```...time_end``` parameter account for the number of frames used for inference and evaluation. We used a value of *3200* for the *WBC-SPH* data set, *600* for *WaterRamps*, and *600* for *Liquid3d*.
The ```...time_end``` parameter account for the number of frames used for inference and evaluation. We used a value of *3200* for the *WBC-SPH* data set, *600* for *WaterRamps*, and *200* for *Liquid3d*.
The generated test files are stored in the ```pipeline.output_dir``` folder, specified in the config file. The output files have a *hdf5* format and can be rendered with the ```utils/draw_sim2d.py``` script.

Rendering of a small sample sequence:
Expand Down
Binary file removed checkpoints/Liquid3d/ckpt-51.data-00000-of-00001
Binary file not shown.
Binary file removed checkpoints/Liquid3d/ckpt-51.index
Binary file not shown.
Binary file modified checkpoints/Liquid3d/ckpt.data-00000-of-00001
Binary file not shown.
Binary file modified checkpoints/Liquid3d/ckpt.index
Binary file not shown.
8 changes: 2 additions & 6 deletions configs/Liquid3d.yml
Original file line number Diff line number Diff line change
Expand Up @@ -60,12 +60,8 @@ pipeline:
#max_err: 0.0025 # 1/2 voxel
max_dens_err: 0.1 # 20% density error

warm_up_bnds: [
20000,
30000,
40000
]
max_warm_up: [0, 2, 3, 5]
warm_up_bnds: []
max_warm_up: [0]

its_bnds: []
iterations: [0]
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -11,4 +11,4 @@ tqdm
sklearn
tensorpack
lapsolver
python-prctl
python-prctl

0 comments on commit 305d41b

Please sign in to comment.