The invention belongs to the technical field of large-scale
deep learning model training, and particularly relates to a
video memory scheduling system based on dynamic sequence length
assembly line parallel training. The
system comprises a dynamic recalculation module, a
video memory arrangement and prefetching module and a mapping reconstruction module, the
system collects the length and load information of each micro-batch before each round of iteration, establishes an optimization model with the goal of minimizing the iteration time, dynamically generates a re-calculation plan, and realizes inter-stage
video memory load balancing; by introducing a collating
stream and an asynchronous scheduler outside a main computing
stream, overlapping execution of video memory collating and computing prefetching is realized, and the influence of fragment collating on
training performance is reduced; a dynamic physical block mapping strategy is designed based on a
CUDA virtual memory management mechanism, the physical block
granularity is adaptively adjusted according to the idle state of the video memory, and the API calling and context switching overhead is reduced. Experimental results show that the video memory
utilization rate and the
system stability are remarkably improved, and an efficient video
memory scheduling scheme is provided for
large model training.