You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.