tomodrgn.config.save_config#
- save_config(args: Namespace, star: TiltSeriesStarfile, data: TiltSeriesMRCData, lat: Lattice, model: TiltSeriesHetOnlyVAE | FTPositionalDecoder | DataParallelPassthrough, out_config: str) None[source]#
Save input arguments and precalculated data, lattice, and model metadata. Called automatically during train_vae and train_nn.
- Parameters:
args – argparse namespace from parse_args
star – TiltSeriesStarfile object describing input data
data – TiltSeriesMRCData object containing input data
lat – Lattice object created using input data
model – TiltSeriesHetOnlyVAE or FTPositionalDecoder object created using input data (no weights included)
out_config – name of output .pkl file in whcih to save configuration
- Returns:
None