Add --save_model flag to the training example

[braswent]

Can we add the --save_model flag to the "train the matching model" example? This will allow users to know how to produce the pt checkpoints to run the matching models in the second example.

An alternate suggestion is to add a link to the example shown on Running the matcher google collab so users know how to access the pt files to run the matcher.