Skip to content

Megatron to nemo checkpointing misses arguments #14363

@JeremieDona

Description

@JeremieDona

Describe the bug
When looking at the page here describing megatron -> nemo checkpoint conversion it points to this scripts

The line device = torch.device("cuda" if torch.cuda.is_available() and not args.no_cuda else "cpu") specifies the parsed argument "no_cuda" which is not defined in the argument parser.

Simply adding the required argument does not fix the script in my case.

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions