Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

general trainer class for GINO and NO #215

Merged
merged 32 commits into from
Sep 22, 2023
Merged

general trainer class for GINO and NO #215

merged 32 commits into from
Sep 22, 2023

Conversation

dhpitt
Copy link
Collaborator

@dhpitt dhpitt commented Sep 9, 2023 •
edited
Loading

This PR extends the functionality of a Trainer to training GINO models.

I will be opening a separate PR for train_CFD and associated functions, because of the o3d dependency.

dhpitt and others added 30 commits September 9, 2023 16:57
@dhpitt
general trainer class for GINO and NO
b09c891
@dhpitt
cfd training setup
d44401c
simple callback examples
3be5fa6
@dhpitt
simple callback examples
731d182
@dhpitt
Merge branch 'gino-trainer' of github.com:dhpitt/neuraloperator into …
fbb09c6 
…gino-trainer
@dhpitt
callback for output encoding
e07c052
@dhpitt
training with callback hooks
899682b
@dhpitt
train_darcy runs with new trainer Callbacks
498afa2
@dhpitt
FNO takes kwargs for trainer callbacks
c63fe29
@dhpitt
fix path typo
a246a49
@dhpitt
train cfd first edits
d70eca1
@dhpitt
Merge branch 'main' into gino-trainer
b2bd7f6
@dhpitt
remove cfd dependence
001d066
@dhpitt
val loss computation callback abstract method
0e2b4de
@dhpitt
wandb logger callback added to train_darcy
8184dff
@dhpitt
cfd training almost works
7f8fde9
@dhpitt
cfd training script
8a021b8
@dhpitt
config for cfd training script
139c0e0
@dhpitt
full featured training config
a8fda4a
@dhpitt
cfd training script with proper path
873a25c
@dhpitt
preprocess cfd for mesh_datamodule
361a7b7
@dhpitt
remove all references to cfd
c543b80
@dhpitt
notimplementederror on base class compute loss
c66bb97
@dhpitt
output encoder callback
8905f5c
@dhpitt
add metalosses, change base callback default to Pass
7eb5589
@dhpitt
CallbackPipeline
5b5eb24
@dhpitt
trainer with CallbackPipeline
fd6cf99
@dhpitt
navier stokes script
c694a06
@dhpitt
Merge branch 'main' into gino-trainer
765870c
@dhpitt
navier stokes with new callbacks
f9628b9
@dhpitt
Copy link
Collaborator Author

dhpitt commented Sep 21, 2023

Trainer performance in train_navier_stokes.py (n_train 2000, n_test 500, resolution 128, n_epochs 100):

Before: 128_h1=0.1132, 128_l2=0.0412

After: 128_h1=0.1133, 128_l2=0.0413

@JeanKossaifi
Copy link
Member

Great changes @dhpitt! Merging.

@JeanKossaifi JeanKossaifi merged commit 187f46f into neuraloperator:main Sep 22, 2023
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@dhpitt @JeanKossaifi