diff --git a/configs/router_energy_n10_embedding.toml b/configs/router_energy_n10_embedding.toml new file mode 100644 index 0000000..fc1bee4 --- /dev/null +++ b/configs/router_energy_n10_embedding.toml @@ -0,0 +1,23 @@ +[train] +mode = "flow" +epochs = 30 +lr = 3e-4 +warmup_epochs = 3 +val_fraction = 0.1 +num_workers = 4 + +[model] +conditioning = "embedding" +dropout = 0.0 +expert_hidden_dim = 128 +expert_n_blocks = 4 + +[model.router] +enabled = true +type = "energy" +n_experts = 10 +expert_hidden_dim = 128 +expert_n_blocks = 4 +temperature = 0.05 +lambda_balance = 0.035 +learn_centers = true diff --git a/configs/router_energy_n10_physical.toml b/configs/router_energy_n10_physical.toml new file mode 100644 index 0000000..ba2a653 --- /dev/null +++ b/configs/router_energy_n10_physical.toml @@ -0,0 +1,23 @@ +[train] +mode = "flow" +epochs = 30 +lr = 3e-4 +warmup_epochs = 3 +val_fraction = 0.1 +num_workers = 4 + +[model] +conditioning = "physical" +dropout = 0.0 +expert_hidden_dim = 128 +expert_n_blocks = 4 + +[model.router] +enabled = true +type = "energy" +n_experts = 10 +expert_hidden_dim = 128 +expert_n_blocks = 4 +temperature = 0.05 +lambda_balance = 0.035 +learn_centers = true diff --git a/configs/wgan_h128_b4_physical.toml b/configs/wgan_h128_b4_physical.toml new file mode 100644 index 0000000..ca6ca87 --- /dev/null +++ b/configs/wgan_h128_b4_physical.toml @@ -0,0 +1,13 @@ +[train] +mode = "wgan" +epochs = 30 +lr = 3e-4 +warmup_epochs = 3 +val_fraction = 0.1 +num_workers = 4 + +[model] +hidden_dim = 128 +n_blocks = 4 +dropout = 0.0 +conditioning = "physical"