Skip to content
Snippets Groups Projects
Select Git revision
  • main default protected
  • training_checkpointing
  • vladb/ddp
  • vladb/test_train
  • vladb/unit_tests
5 results
You can move around the graph by using the arrow keys.
Created with Raphaël 2.2.017Jul10Jun4318Feb161530Jan2928262524221812119653228Dec2724Nov8227OctWIP training checkpointingtraining_checkp…training_checkpointingFix inference codemainmainAdd conda environment file, update READMEAdd DistributedDataParallel trainingAdd logging throughout the framework codeRemove useless models init fileAdd TrainerArguments and adjust training loopDrop Optimus DataLoader, use PyTorch's built-inMove to HuggingFace tokenizersMove to HuggingFace datasetsChange quotes styleAdd gradient checkpointing option to OptimusAdd PyTorch built-in SDPA to OptimusREADME cleanupMove Optimus configuration into separate config classAdapt optimus to Distributonvladb/ddpvladb/ddpIntroduce the module for distributed trainingAdd transformer based on PyTorch TransformerEncoderIntroduce automatic sanity check for trainingvladb/test_trainvladb/test_trainMerge branch 'feature/fp16' into 'main'Adjust optimizer epsilon value for AMPAdd fp16 mixed precision trainingMerge branch 'fix/estimation_interval' into 'main'Basis for unit testingvladb/unit_testsvladb/unit_testsAdd gitlab CI/CD fileFix estimation intervalMerge branch 'feature/inference' into 'main'Add tokens per second informationAdd option to pass a prompt to the inference scriptAdd inference docs to README.mdAdd inference codeRevert "Merge branch 'vladb/add_inference' into 'main'"Merge branch 'feature/folder_structure' into 'main'Merge branch 'vladb/add_inference' into 'main'Update README fileAdd project LICENSERestructure projectMerge branch 'feature/merge_request_template' into 'main'Merge branch 'fix/datasets' into 'main'Merge branch 'fix/general_small_fixes' into 'main'