spyrit
2.3.2
Contents
spyrit.core
spyrit.core.meas
spyrit.core.nnet
spyrit.core.noise
spyrit.core.prep
spyrit.core.recon
spyrit.core.torch
spyrit.core.train
spyrit.core.train.attr_removal
spyrit.core.train.attr_transformation
spyrit.core.train.boxplot
spyrit.core.train.boxplotconsist
spyrit.core.train.checkpoint
checkpoint()
spyrit.core.train.compare_model
spyrit.core.train.count_memory
spyrit.core.train.count_param
spyrit.core.train.count_trainable_param
spyrit.core.train.images_norm
spyrit.core.train.imshow
spyrit.core.train.load_net
spyrit.core.train.multiplot
spyrit.core.train.read_param
spyrit.core.train.remove_model_attributes
spyrit.core.train.rename_model_attributes
spyrit.core.train.save_net
spyrit.core.train.tb_profiler
spyrit.core.train.tb_writer_add_image
spyrit.core.train.tb_writer_add_scalar
spyrit.core.train.tb_writer_init
spyrit.core.train.train_model
spyrit.core.train.train_model_supervised
spyrit.core.train.visualize_conv_layers
spyrit.core.train.visualize_model
spyrit.core.train.Train_par
spyrit.core.train.Weight_Decay_Loss
spyrit.core.warp
spyrit.misc
spyrit.external
Tutorials
Tutorials
spyrit
spyrit.core
spyrit.core.train
spyrit.core.train.checkpoint
Edit on GitHub
spyrit.core.train.checkpoint
spyrit.core.train.
checkpoint
(
root
,
epoch
,
model
)
[source]
Saves the dictionaries of a given pytorch model for the right epoch