Skip to content

Commit

Permalink
CRITICAL: change to five modes in IC
Browse files Browse the repository at this point in the history
  • Loading branch information
Ceyron committed May 24, 2024
1 parent 6c57f93 commit d6c0e27
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion apebench/_base_scenario.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class BaseScenario(eqx.Module, ABC):
num_channels: int = 1

# Settings for both training and testing
ic_config: str = "fourier;10;true;true"
ic_config: str = "fourier;5;true;true"
num_warmup_steps: int = 0

# Setting up the training
Expand Down
2 changes: 1 addition & 1 deletion apebench/scenarios/difficulty/_nonlinear.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ class FisherKPP(Nonlinear):
drag_gamma: float = 0.02
diffusion_gamma: float = 0.2

ic_config: str = "clamp;0.0;1.0;fourier;10;false;false" # Overwrite
ic_config: str = "clamp;0.0;1.0;fourier;5;false;false" # Overwrite

def __post_init__(self):
self.gammas = (self.drag_gamma, 0.0, self.diffusion_gamma)
Expand Down
2 changes: 1 addition & 1 deletion apebench/scenarios/normalized/_nonlinear.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ class FisherKPP(Nonlinear):
drag_alpha: float = 0.02
diffusion_alpha: float = 4e-6

ic_config: str = "clamp;0.0;1.0;fourier;10;false;false" # Overwrite
ic_config: str = "clamp;0.0;1.0;fourier;5;false;false" # Overwrite

def __post_init__(self):
self.alphas = (self.drag_alpha, 0.0, self.diffusion_alpha, 0.0, 0.0)
Expand Down
2 changes: 1 addition & 1 deletion apebench/scenarios/physical/_polynomial.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ class FisherKPP(Polynomial):
diffusion_coef: float = 0.004
quadratic_coef: float = -20.0

ic_config: str = "clamp;0.0;1.0;fourier;10;false;false" # Overwrite
ic_config: str = "clamp;0.0;1.0;fourier;5;false;false" # Overwrite

def __post_init__(self):
self.a_coefs = (self.drag_coef, 0.0, self.diffusion_coef, 0.0, 0.0)
Expand Down

0 comments on commit d6c0e27

Please sign in to comment.