Add option to run.py to only run captioning

[elizlee]

Part of running baseline captioning against one of the datasets identified in Unified-IO.

I only modified run.py such that adding the flag --captions-only should only run the captioning, but I haven't tested this yet.