The invention discloses a two-stage strategy gradient optimization automatic parallel method based on adaptive graph Mangbar, which comprises the following steps of: firstly, acquiring a public AI model
data set, performing operator fusion on each computational graph in the
data set, and acquiring a
feature matrix X of the computational graph; then X is used as an initial coding matrix and is input into a Pitman-Ba neural network to obtain a
feature code, after a feature # imgabs0 # is generated through a multi-layer
perceptron, # imgabs1 # is input into a disturbance two-stage strategy gradient
algorithm, j equipment placement strategies are generated through disturbance
noise in each
training period and internal stage, and parameters involved in the disturbance
noise generation process are updated; and in the external stage, carrying out equipment placement strategy evaluation, selecting a minimum
training time strategy to carry out global parameter updating, and outputting an optimal equipment placement strategy. According to the method, the problem of limited
receptive field of a traditional graph neural network is solved by effectively extracting node features and capturing a long-term dependency relationship, and an optimal equipment parallel strategy is obtained.