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
-
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