Updates to trainer

Now hooks up to new modalities and uses a color mask for no-depth
training.