issues
search
braceal
/
molecules
Machine learning for molecular dynamics.
MIT License
5
stars
5
forks
source link
Feature/multi task
#91
Closed
braceal
closed
3 years ago
braceal
commented
3 years ago
Updates
Bug fix: moved checkpoint log out of VAE train loop.
Updated PointCloudDataset and PointCloudInMemoryDataset to return a dictionary with arbitrary scalars and train/valid data.
Updated AAE train/valid logic to work with new CheckpointCallback interface and with new dictionary dataloader return type.
Black style updates to AAE code including PointCloud Datasets.
Updates