Feature/multi task - Githubissues

braceal / molecules

Machine learning for molecular dynamics.

MIT License

5 stars 5 forks source link

Closed braceal closed 3 years ago

braceal commented 3 years ago

Updates

Generalized CheckpointCallback to save arbitrary checkpoint dictionaries stored in the logs.
Added model property to VAE class to handle DDP model wrapper.
Updated VAE to work with new CheckpointCallback.