Skip to content

Commit

Permalink
update gradio more easy 3
Browse files Browse the repository at this point in the history
  • Loading branch information
lpscr committed Oct 24, 2024
1 parent 0d58148 commit 9ab4e85
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions finetune_gradio.py
Original file line number Diff line number Diff line change
Expand Up @@ -392,7 +392,7 @@ def transcribe_all(name_project, audio_files, language, user=False, progress=gr.
path_project_wavs = os.path.join(path_project, "wavs")
file_metadata = os.path.join(path_project, "metadata.csv")

if user == False:
if not user:
if audio_files is None:
return "You need to load an audio file."

Expand Down Expand Up @@ -625,7 +625,7 @@ def calculate_train(
mini_batch_frames = frames_per_gpu * grad_accum * gpus
mini_batch_hours = mini_batch_frames * mel_hop_length / mel_sampling_rate / 3600
updates_per_epoch = total_hours / mini_batch_hours
steps_per_epoch = updates_per_epoch * grad_accum
# steps_per_epoch = updates_per_epoch * grad_accum
epochs = wanted_max_updates / updates_per_epoch

if finetune:
Expand Down

0 comments on commit 9ab4e85

Please sign in to comment.