add code for training AUGUSTUS from a reference