62 lines
1.2 KiB
TOML
62 lines
1.2 KiB
TOML
|
[run]
|
||
|
|
name = "model_class_frontier_7gb_01_siren_conditioned_inr"
|
||
|
|
seed = 20260725
|
||
|
|
artifact_dir = "artifacts/current_run/training_runs/siren_conditioned_inr"
|
||
|
|
|
||
|
|
[data]
|
||
|
|
root = "data/processed/full"
|
||
|
|
train_cases = 45
|
||
|
|
val_cases = 3
|
||
|
|
test_cases = 2
|
||
|
|
points_per_case = 999999999
|
||
|
|
batch_size = 4096
|
||
|
|
source = "local"
|
||
|
|
|
||
|
|
[model]
|
||
|
|
type = "siren_conditioned_inr"
|
||
|
|
hidden_width = 1024
|
||
|
|
depth = 6
|
||
|
|
activation = "gelu"
|
||
|
|
coordinate_features = ["x", "y", "sdf"]
|
||
|
|
condition_width = 512
|
||
|
|
condition_depth = 3
|
||
|
|
condition_dim = 512
|
||
|
|
siren_omega0 = 30.0
|
||
|
|
|
||
|
|
[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"
|
||
|
|
group = "model_class_frontier_7gb_01"
|
||
|
|
tags = ["airfrans", "model-class-frontier", "7gb-subset", "tarball-data", "siren_conditioned_inr"]
|
||
|
|
|
||
|
|
[huggingface]
|
||
|
|
enabled = true
|
||
|
|
repo_id = "zacheryasc/airfrans-frontier-checkpoints"
|
||
|
|
repo_type = "model"
|
||
|
|
path_prefix = "model_class_frontier_7gb_01/siren_conditioned_inr"
|
||
|
|
private = false
|