Add TODO.
Free C allocated model after train.