dyngpt.solver.pre_training

dyngpt.solver.pre_training(config_pretrain)

Pre-train the neural network model for state transition network.

Args:

config_pretrain (Namespace): Configuration object containing training parameters.

Returns:

dict: A dictionary containing training loss means and the final model weight path.