Adaptive neural network training method, electronic device, medium and program product

The invention provides a self-adaptive neural network training method, electronic equipment, a medium and a program product, and the method comprises the steps: training a target neural network based on a self-adaptive parameter of a current training round, and the self-adaptive parameter is used for determining the training task amount of each training node for training the target neural network; and based on the training time of each training node in the current training round, adjusting the adaptive parameter of the current training round, and determining the adjusted adaptive parameter as the adaptive parameter of the next training round. According to the method, the electronic equipment, the medium and the program product provided by the invention, the training tasks do not need to be manually allocated to the training nodes with different performances, the labor cost can be reduced, the training efficiency of the deep learning task under the heterogeneous cluster can be improved, and meanwhile, the dist