Resource management method and device for distributed machine learning tasks
A machine learning and resource management technology, applied in the field of resource management of distributed machine learning tasks, can solve problems such as memory allocation and performance impact are not considered
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0030] According to an embodiment of the present invention, a resource management method for distributed machine learning tasks is provided, see figure 1 , including the following steps:
[0031] S101: The user submits a machine learning task, which includes two aspects of information, one is the size of the data set, and the other is the number of containers;
[0032] S102: the prediction model calculates the allocation size of the memory according to the size of the data set and the number of containers, and selects a corresponding cache mode;
[0033] S103: Divide the memory allocation into two cases according to the selection of the cache mode. When the memory is sufficient, select the optimal performance model; when the memory is insufficient, select the optimal resource utilization model.
[0034] The resource management method for distributed machine learning tasks in the embodiments of the present invention saves resources and improves task performance through memory ...
Embodiment 2
[0051] According to another embodiment of the present invention, a resource management device for distributed machine learning tasks is provided, see Figure 4 ,include:
[0052] The submission unit 201 is used for users to submit machine learning tasks, which include two aspects of information, one is the size of the data set, and the other is the number of containers;
[0053] The cache mode selection unit 202 is used for the prediction model to calculate the allocation size of the memory according to the data set size and the number of containers, and select the corresponding cache mode at the same time;
[0054] The memory allocation unit 203 is configured to divide the memory allocation into two situations according to the selection of the cache mode. When the memory is sufficient, the optimal performance model is selected; when the memory is insufficient, the optimal resource utilization model is selected.
[0055] The resource management device for distributed machine ...
Embodiment 3
[0066] A storage medium stores program files capable of implementing any one of the resource management methods for the above-mentioned distributed machine learning tasks.
PUM
Login to View More Abstract
Description
Claims
Application Information
Login to View More 


