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