not getting .ckpt file after training #1835
Unanswered
PrernaDev501
asked this question in
Q&A
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 the lighning_logs folder where checkpoints of trained model are saved how can i fix this ?
Beta Was this translation helpful? Give feedback.
All reactions