{ | |
"dim_in": 14, | |
"dim_out": 3, | |
"gradient_checkpointing": false, | |
"hidden_channels": 128, | |
"modes1": 16, | |
"modes2": 16, | |
"modes3": 16, | |
"n_spatial_dims": 2, | |
"spatial_resolution": [ | |
256, | |
256 | |
] | |
} |
{ | |
"dim_in": 14, | |
"dim_out": 3, | |
"gradient_checkpointing": false, | |
"hidden_channels": 128, | |
"modes1": 16, | |
"modes2": 16, | |
"modes3": 16, | |
"n_spatial_dims": 2, | |
"spatial_resolution": [ | |
256, | |
256 | |
] | |
} |