search

ramanathanlab / mdlearn

Machine learning for molecular dynamics
MIT License
12 stars 7 forks source link

CVAE Trainer class #30

Closed braceal closed 2 years ago

braceal commented 3 years ago

Development of a CVAE Trainer class similar to LinearAETrainer for this model: https://github.com/ramanathanlab/mdlearn/blob/main/mdlearn/nn/models/vae/symmetric_conv2d_vae.py