not getting .ckpt file after training #1834
PrernaDev501
started this conversation in
Development
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
from pyannote.database import registry
registry.load_database("/content/drive/MyDrive/Jatayu_internship/Pyannote/database.yml")
ami = registry.get_protocol('SRM.SpeakerDiarization.mini')
from pyannote.audio.tasks import VoiceActivityDetection
vad_task = VoiceActivityDetection(ami, duration=2.0, batch_size=32)
from pyannote.audio.models.segmentation import PyanNet
model = PyanNet(task=vad_task, sincnet={'stride': 20})
import pytorch_lightning as pl
trainer = pl.Trainer(devices=1, max_epochs=1)
trainer.fit(model)
I am training vad model using this script . After training i am not getting checkpoints file(.ckpt) . How can i fix this
Beta Was this translation helpful? Give feedback.
All reactions