NVIDIA / NVIDIA/Megatron-LM

[QUESTION] inprocess restart

Open
#2,774 5 comments 0 reactions 1 assignee Claimed by @ahmadki View on GitHub
community-request question waiting-on-customer
Dominant language
Python
Stars
17.9k
Forks
4.5k
Avg merge
4d 6h
Merged PRs (30d)
271

Description

Hi,
I added some code in training.py file to inject fault as below. And add the config --inprocess-restart.

`

if iteration == 5 and torch.distributed.get_rank() == 2:
from nvidia_resiliency_ext.inprocess.tools.inject_fault import inject_fault, Fault
inject_fault((Fault.GPU_SLEEP), 4, 1, 30, 1)

`

I saw that when my training work run to the 5th iteration, the process-restart will make the training work run from the first iteration again, without freeing the GPU memory.
So my question is:
1. Did i use process-restart right?
2. Why process-restart not release the previous memory . If the training work meet out of memory error, the training work will fall into a infinite loop.
3. If the training work run from the first iteration without saving the ckpt. This will waste previous training results?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.