Skip to content

Commit

Permalink
add scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
ziyiyin97 committed May 21, 2024
1 parent 115b72a commit fb9dd99
Show file tree
Hide file tree
Showing 3 changed files with 630 additions and 2 deletions.
7 changes: 6 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,15 @@ First, install [Julia](https://julialang.org/) and [Python](https://www.python.o

## Scripts

[wiser.jl](scripts/gen_cig_openfwi.jl) runs the WISER algorithm in the paper to perform physics-based latent space correction.
[wiser.jl](scripts/wiser.jl) runs the WISER algorithm in the paper to perform physics-based latent space correction.

The script [utils.jl](scripts/utils.jl) parses the input as keywords for each experiment.

The following keyword arguments can be used to reproduce the results in the WISER paper:

- Case 1: `julia wiser.jl --lr_wiser=0.004`
- Case 2: `julia wiser.jl --test_snr=0.0 --amplitude=0.2 --lambda=10.0 --lr_pre=0.0004`

## LICENSE

The software used in this repository can be modified and redistributed according to [MIT license](LICENSE).
Expand Down
2 changes: 1 addition & 1 deletion scripts/utils.jl
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ function parse_commandline()
"--wiser_batch_size"
help = "number of particles during WISER"
arg_type = Int
default = 8
default = 16
"--lambda"
help = "lambda in WISER"
arg_type = Float32
Expand Down
Loading

0 comments on commit fb9dd99

Please sign in to comment.