airfRANS-model-exploration/configs/aggressive_smoke.toml

53 lines
900 B
TOML
Raw Normal View History

[run]
name = "aggressive_smoke"
seed = 20260723
artifact_dir = "artifacts/current_run/training_runs"
[data]
root = "data/processed/full"
train_cases = 45
val_cases = 3
test_cases = 2
points_per_case = 999999999
batch_size = 4096
[model]
type = "film_fourier_mlp"
hidden_width = 4096
depth = 12
activation = "gelu"
coordinate_features = ["x", "y", "sdf"]
fourier_scales = [1.0, 2.0, 4.0, 8.0, 16.0, 32.0]
condition_width = 1024
condition_depth = 3
condition_dim = 512
[optim]
lr = 0.0001
weight_decay = 0.0001
steps = 5000
log_interval = 500
[device]
type = "cuda"
allow_cpu_fallback = false
benchmark_kernels = true
[loss]
type = "normalized_mse"
[precision]
dtype = "bf16"
[checkpoint]
interval_seconds = 1800
[stability]
max_grad_norm = 1.0
[observability]
backend = "wandb"
entity = "zacheryasc-personal"
project = "airfRANS-model-sweep"
tags = ["airfrans", "remote", "aggressive-smoke"]