| seed: 0 | |
| data: | |
| directory: data | |
| filename: Burgers.mat | |
| train_samples: 600 | |
| validation_samples: 200 | |
| test_samples: 200 | |
| space_stride: 32 | |
| time_stride: 10 | |
| model: | |
| grid_size: 33 | |
| features: [64, 128, 128, 256] | |
| latent_channels: 4 | |
| fourier_features: 32 | |
| velocity_widths: [64, 64, 64] | |
| padding_size: 2 | |
| alpha: 1.0e-5 | |
| sobolev_reg: 1.0e-8 | |
| bc_reg: 10.0 | |
| step_size: 0.1 | |
| training: | |
| epochs: 5000 | |
| batch_size: 32 | |
| learning_rate: 0.001 | |
| weight_decay: 0.0 | |
| scheduler_frequency: 250 | |
| scheduler_gamma: 0.9 | |
| validation_frequency: 1000 | |
| checkpoint_frequency: 100 | |
| jit_flow: true | |
| device: cuda | |
| num_threads: 8 | |
| paths: | |
| results: results | |
| checkpoint: weight/best_model.pth | |
| last_checkpoint: weight/last_model.pth | |
| inference: | |
| batch_size: 32 | |
| device: cuda | |
| visualization: | |
| sample_indices: [0, 80, 160] | |
| dpi: 200 | |