diff --git a/README.md b/README.md index 494ad536c2bd3566e88f8a2ef45c5f4b70455075..ccf43a55708e7d6af41437da02d28589b817072e 100644 --- a/README.md +++ b/README.md @@ -35,6 +35,9 @@ DeepSparkHub甄选上百个应用算法和模型,覆盖AI和通用计算各领 | [Llama2-34B](nlp/llm/llama2-34b/pytorch) | PyTorch | Megatron-DeepSpeed | Bookcorpus | 3.4.0 | | [Llama3-8B](nlp/llm/llama3_8b/pytorch) | PyTorch | Megatron-DeepSpeed | Bookcorpus | 4.1.1 | | [Llama3-8B SFT](nlp/llm/llama3_8b_sft/pytorch) | PyTorch | ColossalAI | school_math_0.25M | 4.1.1 | +| [Llama3-8B PPO](nlp/llm/llama3_8b/openrlhf) | PyTorch | OpenRLHF | Llama-3-8b-sft-mixture | 4.2.0 | +| [Llama3-8B DPO](nlp/llm/llama3_8b/openrlhf) | PyTorch | OpenRLHF | Llama-3-8b-sft-mixture | 4.2.0 | +| [Llama3-8B KTO](nlp/llm/llama3_8b/openrlhf) | PyTorch | OpenRLHF | Llama-3-8b-sft-mixture | 4.2.0 | | [Mamba-2](nlp/llm/mamba-2/pytorch) | PyTorch | Megatron-LM | GPT Small-117M | 4.1.1 | | [MiniCPM](nlp/llm/minicpm/pytorch) | PyTorch | DeepSpeed | MiniCPM-2B-sft-bf16 | 4.2.0 | | [Mixtral 8x7B](nlp/llm/mixtral/pytorch) | PyTorch | Megatron-LM | GPT Small-117M | 4.1.1 | @@ -43,6 +46,8 @@ DeepSparkHub甄选上百个应用算法和模型,覆盖AI和通用计算各领 | [QWen1.5-7B](nlp/llm/qwen1.5-7b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [QWen1.5-14B](nlp/llm/qwen1.5-14b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/pytorch) | PyTorch | LLaMA-Factory | qwen2.5-7b | 4.1.1 | +| [Qwen2.5-1.5B verl](nlp/llm/qwen2.5-1.5b/verl) | PyTorch | verl | qwen2.5-1.5b | 4.2.0 | +| [Qwen2.5-7B verl](nlp/llm/qwen2.5-7b/verl) | PyTorch | verl | qwen2.5-7b | 4.2.0 | | [Yi-6B](nlp/llm/yi-6b/pytorch) | PyTorch | DeepSpeed | Yi-6B | 4.2.0 | | [Yi-1.5-6B](nlp/llm/yi-1.5-6b/pytorch) | PyTorch | DeepSpeed | Yi-1.5-6B | 4.2.0 | | [Yi-VL-6B](nlp/llm/yi-vl-6b/pytorch) | PyTorch | LLaMA-Factory | Yi-VL-6B-hf | 4.2.0 | @@ -206,6 +211,7 @@ DeepSparkHub甄选上百个应用算法和模型,覆盖AI和通用计算各领 | [YOLOv8](cv/detection/yolov8/pytorch) | PyTorch | COCO | 3.0.0 | | [YOLOv9](cv/detection/yolov9/pytorch) | PyTorch | COCO | 4.1.1 | | [YOLOv10](cv/detection/yolov10/pytorch) | PyTorch | COCO | 4.1.1 | +| [YOLOv11](cv/detection/yolov11/pytorch) | PyTorch | COCO | 4.2.0 | #### 三维目标检测 diff --git a/README_en.md b/README_en.md index 255003a013cea296cdd29e159cbbcd64140c9be2..3f3d59f009c8c3db734706f7a062183176ff57df 100644 --- a/README_en.md +++ b/README_en.md @@ -37,6 +37,9 @@ individuals, healthcare, education, communication, energy, and more. | [Llama2-34B](nlp/llm/llama2-34b/pytorch) | PyTorch | Megatron-DeepSpeed | Bookcorpus | 3.4.0 | | [Llama3-8B](nlp/llm/llama3_8b/pytorch) | PyTorch | Megatron-DeepSpeed | Bookcorpus | 4.1.1 | | [Llama3-8B SFT](nlp/llm/llama3_8b_sft/pytorch) | PyTorch | ColossalAI | school_math_0.25M | 4.1.1 | +| [Llama3-8B PPO](nlp/llm/llama3_8b/openrlhf) | PyTorch | OpenRLHF | Llama-3-8b-sft-mixture | 4.2.0 | +| [Llama3-8B DPO](nlp/llm/llama3_8b/openrlhf) | PyTorch | OpenRLHF | Llama-3-8b-sft-mixture | 4.2.0 | +| [Llama3-8B KTO](nlp/llm/llama3_8b/openrlhf) | PyTorch | OpenRLHF | Llama-3-8b-sft-mixture | 4.2.0 | | [Mamba-2](nlp/llm/mamba-2/pytorch) | PyTorch | Megatron-LM | GPT Small-117M | 4.1.1 | | [MiniCPM](nlp/llm/minicpm/pytorch) | PyTorch | DeepSpeed | MiniCPM-2B-sft-bf16 | 4.2.0 | | [Mixtral 8x7B](nlp/llm/mixtral/pytorch) | PyTorch | Megatron-LM | GPT Small-117M | 4.1.1 | @@ -45,6 +48,8 @@ individuals, healthcare, education, communication, energy, and more. | [QWen1.5-7B](nlp/llm/qwen1.5-7b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [QWen1.5-14B](nlp/llm/qwen1.5-14b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/pytorch) | PyTorch | LLaMA-Factory | qwen2.5-7b | 4.1.1 | +| [Qwen2.5-1.5B verl](nlp/llm/qwen2.5-1.5b/verl) | PyTorch | verl | qwen2.5-1.5b | 4.2.0 | +| [Qwen2.5-7B verl](nlp/llm/qwen2.5-7b/verl) | PyTorch | verl | qwen2.5-7b | 4.2.0 | | [Yi-6B](nlp/llm/yi-6b/pytorch) | PyTorch | DeepSpeed | Yi-6B | 4.2.0 | | [Yi-1.5-6B](nlp/llm/yi-1.5-6b/pytorch) | PyTorch | DeepSpeed | Yi-1.5-6B | 4.2.0 | | [Yi-VL-6B](nlp/llm/yi-vl-6b/pytorch) | PyTorch | LLaMA-Factory | Yi-VL-6B-hf | 4.2.0 | @@ -208,6 +213,7 @@ individuals, healthcare, education, communication, energy, and more. | [YOLOv8](cv/detection/yolov8/pytorch) | PyTorch | COCO | 3.0.0 | | [YOLOv9](cv/detection/yolov9/pytorch) | PyTorch | COCO | 4.1.1 | | [YOLOv10](cv/detection/yolov10/pytorch) | PyTorch | COCO | 4.1.1 | +| [YOLOv11](cv/detection/yolov11/pytorch) | PyTorch | COCO | 4.2.0 | #### 3D Object Detection diff --git a/RELEASE.md b/RELEASE.md index 3e205d319523f4205699891e16f23d882fbc6c27..500877531bb5065634eb21dadd0d5457e6507612 100644 --- a/RELEASE.md +++ b/RELEASE.md @@ -1,5 +1,54 @@ # DeepSparkHub Release Notes +## DeepSparkHub 25.06 Release Notes + +### 模型与算法 + +* 新增了1个PyTorch训练模型 +* 新增了5个大模型训练示例,涉及verl和OpenRLHF框架 + +
PyTorch | ||||
---|---|---|---|---|
YOLOv11 | ++ | + | ||
大模型 | ||||
Llama3-8B PPO(OpenRLHF) | +Llama3-8B DPO(OpenRLHF) | +Llama3-8B KTO(OpenRLHF) | +||
Qwen2.5-1.5B(verl) | +Qwen2.5-7B(verl) | ++ |