Converts a keras dataset into a tf checkpoint.
E.g.
$ bazel run lingvo/tools:keras2ckpt -- --dataset=mnist