Hugging Face's logo

Spaces:
LorenzoNava
/
deberta-cwe-training
Paused

App Files Community
deberta-cwe-training / train.py

Commit History

feat: Add K-Fold Cross-Validation to train.py
0af6aee

LorenzoNava commited on

feat: Auto-push trained model to HuggingFace Hub
4d107ec

LorenzoNava commited on

feat: Use cleaned dataset (local or HuggingFace)
152f94b

LorenzoNava commited on

feat: Use local cleaned dataset (only standard CWE-XXXX IDs)
006ab48

LorenzoNava commited on

fix: Disable gradient checkpointing for 4x L4 GPUs - fixes backward pass error
1dd8e1d

LorenzoNava commited on

feat: Add standalone training script without Gradio UI
8d7694e

LorenzoNava commited on