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
I have multiple checkpoints produced after running examples/nlp/language_modeling/megatron_gpt_continue_training.py.
However, I am unable to use examples/nlp/language_modeling/megatron_ckpt_to_nemo.py to convert it to a .nemo object. It's probably because the environment in which I want to do the conversion is the not the same as the one used for training. Is there some way to do the conversion only on CPU or just 1 GPU?
I have tried using two different sets of parameters:
Hi @dachenlian,
Could you try to specify --hparams_yaml param? You can find hparams.yaml file in train results folder. It should be in "version_0" or "run_0" folder.
I have multiple checkpoints produced after running
examples/nlp/language_modeling/megatron_gpt_continue_training.py
.However, I am unable to use
examples/nlp/language_modeling/megatron_ckpt_to_nemo.py
to convert it to a.nemo
object. It's probably because the environment in which I want to do the conversion is the not the same as the one used for training. Is there some way to do the conversion only on CPU or just 1 GPU?I have tried using two different sets of parameters:
Input:
Output:
Input:
Output:
The text was updated successfully, but these errors were encountered: