sponsorblock-ml / src /train.py

Commit History

Initialize logging in each script
c4f250e

Joshua Lochner commited on

Revert model input size back to 512 tokens
721bf64

Joshua Lochner commited on

Fix conflicting `--no_cuda` argument
09cabec

Joshua Lochner commited on

Add `no_cuda` argument to not use GPU
de9c8c4

Joshua Lochner commited on

Move `load_datasets` to train script
086ca93

Joshua Lochner commited on

Remove unused imports
bdfb4b1

Joshua Lochner commited on

Remove unused code in training script
31d605f

Joshua Lochner commited on

Change to multiclass classifier
320a2ba

Joshua Lochner commited on

Add functionality to predict self-promo and interaction reminders
90d1f68

Joshua Lochner commited on

Fix prediction and evaluation arguments
9b9ffd0

Joshua Lochner commited on

Add source code
5fbdd3c

Joshua Lochner commited on