deeperwin.optimization.pretrain_orbitals

deeperwin.optimization.pretrain_orbitals(orbital_func, mcmc_state: deeperwin.mcmc.MCMCState, params, fixed_params, config: deeperwin.configuration.PreTrainingConfig, phys_config: deeperwin.configuration.PhysicalConfig, model_config: deeperwin.configuration.ModelConfig, rng_seed: int, loggers: Optional[deeperwin.loggers.DataLogger] = None, opt_state=None)