How do I pass a NEMO model for training? Or does the framework only supports HF.
Open
bug
community-request
external
t-pytdensor
waiting-on-customer
x-sarvam
- Dominant language
- Python
- Stars
- 2k
- Forks
- 561
- Avg merge
- 4d 5h
- Merged PRs (30d)
- 145
Description
This is how my nemo model looks like:
```
/path/to/runs/tts/model
│
├── lightning_logs/
├── model/
├── model--reduced_train_loss=4.0176-epoch=1-consumed_samples=91406336.0-last/
│ ├── context/
│ │ ├── /
│ │ ├── /
│ │ ├── /
│ │ ├── /
│ │ ├── io.json
│ │ └── model.yaml
│ │
│ └── weights/
│ ├── __0_0.distcp
│ ├── __0_1.distcp
│ ├── __1_0.distcp
│ ├── __2_0.distcp
│ ├── ... (many distcp shard files)
│ ├── __127_1.distcp
│ ├── common.pt
│ └── metadata.json
│
└── wandb/
```
I policy can I pass this somehow?
Contributor guide
Assessment
This issue has not been assessed yet.