data_root: folder containing image datasetlogging_root: folder to save all experiments intrain_test: specifytrainorvalto specify which dataset to loadexp_name: name of experiment to specifycheckpoint: load checkpoint file if you want to continue training from a previous training sessionmax_epoch: maximum number of epochslr: learning ratebatch_size: batch sizereg_weight: regularizer factorinit_K: initialization Wiener filteringuse_weiener:trueorfalseto learn wiener filtering damping factor Kresolution: int for resolution of point spread functionpixel_pitch: float for pixel sizefocal_length: float for distance between phase mask and sensorr_cutoff:nullor an int for the radius of the aperturerefractive_idc: refractive index of phase maskwavelength: float for wavelength of interestinit_lens:random,fresnel,planofor initializing the height map of the phase masksingle_image: boolean for testing optimizing phase mask over one imagedownload_data: boolean for downloading data (should befalseafter first download)