Spatial tile-based training for memory efficiency#37

Draft

forklady42 wants to merge 9 commits intohanaol/setup-ptlightningfrom

Commits on Nov 13, 2025

refactored the code with PyTorch Lightning
Hananeh Oliaei
committed
job scripts for training the model on either QM9 or MP data
Hananeh Oliaei
committed

Commits on Nov 14, 2025

updated the dataloader and config scripts according
Hananeh Oliaei
committed

Commits on Nov 18, 2025

added sample grid_half data/input charge densities
Hananeh Oliaei
committed

Commits on Dec 2, 2025

feat: add spatial patch-based training for memory efficiency

forklady42
and
claude
committed

Commits on Dec 8, 2025

Keep validation on GPU
forklady42
committed

Commits on Dec 9, 2025