hatespeech / train.py

Commit History

Add onnxscript dependency and set dynamo=False for classic ONNX export
5097547

Isa0 commited on

Configure dropout via AutoConfig dynamically to support DistilBERT
4aaffa6

Isa0 commited on

Fix optim import and ensure PyTorch AMP GradScaler backwards compatibility
95ead03

Isa0 commited on

Add PyTorch training pipeline with ONNX/INT8 export and inference script
57700c9

Isa0 commited on