summaryrefslogtreecommitdiff
path: root/main.py
diff options
context:
space:
mode:
authorvin <git@vineetk.net>2025-08-24 00:31:29 -0400
committervin <git@vineetk.net>2025-08-24 00:31:29 -0400
commit0f5fcee95648d706b792affee0658e2e652fd904 (patch)
tree68d358b64d76fdb6613f37ff6953a00694bcc2a1 /main.py
parent48cdc65c0eafa08fc39b5bfa49ef7c42943ac91d (diff)
try finetuning encoder-decoder model like t5gemmaHEADmaster
Unfortunately once it finishes the finetuning process, it doesn't actually output anything. With decoder-only models like small gemma3, it also doesn't output anything meaningful. I'm tempted to just classify each token as an ad or not, finetune an encoder like some long-context BERT, and then a post-processing script to convert back to tokens and their timestamps. May end up being more performant and efficient too...
Diffstat (limited to 'main.py')
-rw-r--r--main.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/main.py b/main.py
index a58e02f..6887ac3 100644
--- a/main.py
+++ b/main.py
@@ -401,7 +401,7 @@ def process_transcript_json(input_json_path: str, output_jsonl_path: str):
401 401
402 final_results.append({ 402 final_results.append({
403 "id": prompt_id, 403 "id": prompt_id,
404 "text": full_prompt_for_finetuning, 404 "input": full_prompt_for_finetuning,
405 "target": response_text.strip() 405 "target": response_text.strip()
406 }) 406 })
407 407