diff --git a/configs/wgan_h512_b6_physical.toml b/configs/wgan_h512_b6_physical.toml new file mode 100644 index 0000000..2a5501c --- /dev/null +++ b/configs/wgan_h512_b6_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 = 512 +n_blocks = 6 +dropout = 0.0 +conditioning = "physical"