OLMo icon indicating copy to clipboard operation
OLMo copied to clipboard

integrate mock vision backbone into model

Open epwalsh opened this issue 1 year ago • 2 comments

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

epwalsh avatar Feb 08 '24 23:02 epwalsh