File size: 823 Bytes
e456924 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | 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
|