MODEL-AWARE METHOD AND SYSTEM FOR TRAINING AND/OR FINE-TUNING A MACHINE LEARNING MODEL

    公开(公告)号:US20230289616A1

    公开(公告)日:2023-09-14

    申请号:US18319636

    申请日:2023-05-18

    Applicant: Lemon Inc.

    CPC classification number: G06N3/098

    Abstract: System and method of training a machine learning model on a plurality of devices in parallel are provided. The method includes performing a model profiling execution before a model normal execution, allocating tensors of the model into a plurality of chunks based on profiling results from the model profiling execution, and performing the model normal execution on the plurality of devices in parallel to train or fine-tune the model.

Patent Agency Ranking