OLMo
OLMo copied to clipboard
integrate mock vision backbone into model
Adds an abstraction for vision backbones, along with a mock dataset to test it. You can run a test training job on 2 GPUs with the mock dataset like so:
torchrun --nproc-per-node=2 scripts/train.py configs/mm-tiny.yaml