Add bigger WGAN config (hidden_dim=512, n_blocks=6)
Scaled-up variant of the already-trained wgan_h128_b4_physical.toml benchmark config, same training recipe (mode/epochs/lr/warmup), increased model capacity. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
@@ -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"
|
||||||
Reference in New Issue
Block a user