| fno_config: | |
| datapipe: | |
| source: | |
| data_dir: "./data" | |
| file_name: "2D_DarcyFlow_beta0.1_Train.hdf5" | |
| verbose: true | |
| data: | |
| single_file: true | |
| reduced_resolution: 1 | |
| reduced_resolution_t: 1 | |
| reduced_batch: 1 | |
| initial_step: 1 | |
| test_ratio: 0.1 | |
| pde_name: "2D_Darcy_Flow" | |
| dataloader: | |
| batch_size: 2 | |
| num_workers: 0 | |
| pin_memory: false | |
| model: | |
| name: "FNO" | |
| num_channels: 1 | |
| modes: 4 | |
| width: 8 | |
| training: | |
| output_dir: "./weight" | |
| if_training: true | |
| continue_training: false | |
| model_path: null | |
| save_period: 1 | |
| seed: 0 | |
| training_type: "single" | |
| t_train: 2 | |
| epochs: 1 | |
| optimizer: | |
| name: "Adam" | |
| lr: 1.0e-3 | |
| weight_decay: 1.0e-4 | |
| scheduler: | |
| name: "StepLR" | |
| step_size: 100 | |
| gamma: 0.5 | |
| inference: | |
| checkpoint: "./weight/best_model.pt" | |
| output_dir: "./result/output" | |
| max_batches: 2 | |
| result: | |
| prediction_dir: "./result/output" | |
| output_dir: "./result" | |