File size: 1,254 Bytes
371b59f | 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 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | root:
common:
task: "forward"
device: "auto"
dtype: "float32"
seed: 1111
weight_dir: "weight"
result_dir: "result"
data:
mat_file: "data/AC.mat"
n_train: 8000
evaluation_batch_size: 8192
model:
hidden_layers: [200, 200, 200, 200]
linear_dim: 10
fuzzy_dim: 4
activation: "tanh"
tasks:
forward:
training:
epochs: 10000
lr: 0.001
lbfgs_iters: 0
log_interval: 500
loss:
data: 1.0
pde: 1.0
pde:
lambda_1: 0.0001
lambda_2: 5.0
output:
checkpoint_name: "fpinn_forward.pt"
history_name: "fpinn_forward_history.npz"
predictions_name: "fpinn_forward.npz"
figure_name: "fpinn_forward.png"
inverse:
training:
epochs: 10000
lr: 0.001
lbfgs_iters: 0
log_interval: 500
loss:
data: 1.0
pde: 1.0
pde:
initial_lambda_1: 1.0
initial_lambda_2: 0.0
true_lambda_1: 0.0001
true_lambda_2: 5.0
output:
checkpoint_name: "fpinn_inverse.pt"
history_name: "fpinn_inverse_history.npz"
predictions_name: "fpinn_inverse.npz"
figure_name: "fpinn_inverse.png"
|