scvi.model.base.PyroJitGuideWarmup.on_configure_sharded_model

PyroJitGuideWarmup.on_configure_sharded_model(trainer, pl_module)

Called before configure sharded model

Return type

NoneNone