-
Draft: WIP Training checkpointing 1 of 9 checklist items completed!26
-
Re-factor optimus-prime code (optimus-prime v2) 16 of 31 checklist items completed!25
-
Fix datasets 5 of 9 checklist items completed!24
-
Switch to PyTorch Dataloader and HF datasets 4 of 9 checklist items completed!23
-
Draft: Training plots/graphs + rudimentary model checkpointing after every epoch 3 of 9 checklist items completed!22
-
Draft: Add support for data parallelism on a single node 1 of 9 checklist items completed!21
-
Draft: Introduce automatic sanity check for training 5 of 9 checklist items completed!20
-
Add transformer based on PyTorch TransformerEncoder 5 of 9 checklist items completed!19
-
Draft: Basis for unit testing 5 of 9 checklist items completed!18
-
Add fp16 mixed precision training 5 of 9 checklist items completed!17
-
Fix estimation interval 5 of 9 checklist items completed!16
-
Add inference code 5 of 9 checklist items completed!15
-
Add inference code 5 of 9 checklist items completed!14