add option to save all trained replicates