Minor updates on DDStore and remaining time checking

ORNL / HydraGNN

Distributed PyTorch implementation of multi-headed graph convolutional neural networks

BSD 3-Clause "New" or "Revised" License

61 stars 27 forks source link

Closed jychoi-hpc closed 4 months ago

jychoi-hpc commented 5 months ago

This includes a few updates:

Update on DDStore epoch begin/end routine
Adding check_remaining time routine within a job allocation on Frontier and Perlmutter

allaffa commented 4 months ago

@jychoi-hpc I think there are black formatting incompatibilities that cause the CI tests to crash