ogbg: preload dataset to increase processing speed, cut training time from 3h+ to 1-2h#837

copybara-service[bot] merged 1 commit intomastergoogle/init2winit:masterfrom

test_862981106google/init2winit:test_862981106Copy head branch name to clipboard