feat: add Gradio progress bar and streaming log updates during training 9364099 rshakked commited on May 13
fix: pass tokenizer explicitly to AbuseDataset and safeguard evaluation step 5c16708 rshakked commited on May 13
feat(app): run training on button click and display logs after completion 2032430 rshakked commited on May 13
fix: reduce batch size and enable gradient checkpointing to prevent GPU OOM crashes fedc8f2 rshakked commited on May 13
chore(debug): move device check to top of script for clearer startup logs 285a433 rshakked commited on May 12
chore(debug): print tensor devices and CUDA availability for troubleshooting d4ac0ac rshakked commited on May 12
fix: remove .to(device) from dataset to avoid pin_memory error in Trainer 92846e1 rshakked commited on May 12
updated train_abuse_model to save the traind model after training process 5524706 rshakked commited on May 12
updated app.py to show live outputs and updated gitignore to ignore excel and doc files a9706bf rshakked commited on May 12