awslabs / awslabs/graphstorm

Support end of training in the middle of an epoch

Open
#1,222 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
452
Forks
76
PR merge metrics
No merged PRs in 30d

Description

For very large graphs, training one epoch may take a very long time. People may want to define the number of total iterations for a job instead of the number of epochs.

Contributor guide

Open the contributing guide

Research direction

No file, test, or entry point is named. Start by locating the training configuration and epoch-based training loop, then trace how progress and stopping are handled. Done means a job can specify total iterations, stop in the middle of an epoch, and preserve the existing epoch-based behavior.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.