Skip to content

Commit 9715b33

Browse files
authored
Update README.md
1 parent 7d99f48 commit 9715b33

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

the_well/benchmark/models/fno/README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,10 @@ For benchmarking on the Well, we used the following parameters.
3737
| Blocks | 4 |
3838
| Hidden Size | 128 |
3939

40+
## Information about benchmarking
41+
42+
The models have been trained for a fixed time of 12 hours on a NVIDIA H100 96GB GPU or 500 epochs (whichever happens first). In the time dimension, the context length was set to 4. The batch size was set to maximize the memory usage. We experiment with 5 different learning rates for each model on each dataset, and use the model performing best on the validation set to report test set results (using the `recent.pt` checkpoint). The reported results are here to provide a simple baseline and **should not be considered as state-of-the-art.** We hope that the community will build upon these results to develop better architectures for PDE surrogate modeling.
43+
4044
## Trained Model Versions
4145

4246
Below is the list of checkpoints available for the training of FNO on different datasets of the Well.

0 commit comments

Comments
 (0)