Visualize the generated mixtures. regression_fitting: The heart of RegMix. Fit the regression model using small model performance data, and by default we use the validation loss on the Pile-CC dataset ...