MP_PDE / config.json
yushuang88's picture
Update model configuration
d15f55c verified
Raw
History Blame Contribute Delete
432 Bytes
{
"time_window": 25,
"hidden_dim": 164,
"message_passing_layers": 6,
"neighbor_offsets": [
-3,
-2,
-1,
1,
2,
3
],
"aggregation": "sum",
"instance_norm_affine": false,
"decoder": {
"middle_channels": 8,
"kernel_sizes": [
16,
26
],
"strides": [
3,
1
]
},
"scaling": {
"coordinates": true,
"parameters": true,
"solution": "none"
}
}