Training¶

run_training(config, data_pipeline, sr_model, plotting_interval=1)[source]¶

Run training session with initialized DataPipeline and SRModel.

Parameters

config – Initialized SimpleSR.ConfigUtil object.
data_pipeline – Initialized SimpleSR.DataPipeline object.
sr_model – Initialized SimpleSR.SRModel object.
plotting_interval –

Interval of epochs to plot and save images.

Every batch of the test set will be plotted (interpolated and upsampled by model respectively) and saved. Additionally one batch of training and validation set will be plotted as well.

Note

Plotting can be helpful to monitor progress during training, but will impact performance and therefore lead to longer epoch times.