Example training script for SpanGroupClassifier

allenai / mmda

multimodal document analysis

Apache License 2.0

158 stars 18 forks source link

Open kyleclo opened 2 years ago

kyleclo commented 2 years ago

[ ] SMashed Tokenizer needs to handle expansion of labels to subwords
[ ] Unpacking mapper needs to handle labels
[ ] example building a HFDataset or PTL Trainer that works off mmda.Document export to JSON (this gives token units that are labeled)
[ ] train / save