lora显存占用计算器

开发中

POST

/model/lora_memory_calculator

请求参数

Body 参数application/json

示例

{
    "model_size_in_billion":15,
    "lora_rank":16,
    "sequence_length":55,
    "batch_size":1,
    "gradient_checkpointing":true,
    "optimizer":"adamw",
    "precision":"fp16"

}

请求示例代码

Shell

JavaScript

Java

Swift

PHP

Python

HTTP

Objective-C

Ruby

OCaml

Dart

curl --location --request POST '/model/lora_memory_calculator' \
--header 'Content-Type: application/json' \
--data-raw '{
    "model_size_in_billion":15,
    "lora_rank":16,
    "sequence_length":55,
    "batch_size":1,
    "gradient_checkpointing":true,
    "optimizer":"adamw",
    "precision":"fp16"

}'

返回响应

🟢200成功

application/json

Body

示例

{"status":0,"data":{"memoryUsage":98.4,"gpuNum":2}}

修改于 2025-08-06 14:20:50

基础模型训练显存占用计算器

获取模型详细信息