大模型 | ||||
---|---|---|---|---|
GLM-4 | MiniCPM(DeepSpeed) | Phi-3 | ||
MoE-LLaVA-Phi2-2.7B | MoE-LLaVA-Qwen-1.8B | MoE-LLaVA-StableLM-1.6B | ||
Yi-6B (DeepSpeed) | Yi-1.5-6B (DeepSpeed) | Yi-VL-6B (LLaMA-Factory) |
DeepSparkHub 25.03对应天数软件栈4.2.0版本。
感谢以下社区贡献者
李一力,sanghui-ilu,majorli6,honglyua,吴永乐。
欢迎以任何形式为DeepSparkHub项目贡献。
● 新增了4个PyTorch训练模型,其中2个为Multimodal模型
● 新增了5个大模型训练示例,涉及ColossalAI,Megatron-LM和LLaMA-Factory工具箱
PyTorch | ||||
---|---|---|---|---|
KAN | Mamba-YOLO | LLaVA 1.5 | ||
Stable Diffusion 3 | ||||
大模型 | ||||
DeepSeek MoE 7B (ColossalAI) | Llama3-8B SFT (ColossalAI) | Mamba-2 (Megatron-LM) | ||
Mixtral 8x7B (Megatron-LM) | Qwen2.5-7B SFT (LLaMA-Factory) |
● 更新了Llama2-7B (Megatron-DeepSpeed)模型的数据集链接。
● 同步了DBNet、Fast R-CNN和GoogLeNet等PyTorch模型的代码。
● 同步了BERT Base、FaceNet和ResNet50等TensorFlow模型的代码。
● 同步了ClossalAI、Megatron-DeepSpeed和OpenPCDet工具箱的代码。
DeepSparkHub 24.12对应天数软件栈4.1.1版本。
感谢以下社区贡献者
李一力,sanghui-ilu,majorli6,honglyua,吴永乐。
欢迎以任何形式为DeepSparkHub项目贡献。
PyTorch | ||||
---|---|---|---|---|
Stable Diffusion 1.5 | Stable Diffusion 2.1 | Stable Diffusion XL | ||
CenterMask2 | YOLOv10 |
DeepSparkHub 24.09对应天数软件栈4.1.1版本。
感谢以下人员做出的贡献:
李一力,sanghui-ilu,majorli,Jino Yang,吴永乐。
欢迎以任何形式为DeepSparkHub项目贡献。
PyTorch | ||||
---|---|---|---|---|
Part-A2-Anchor | Part-A2-Free | PointRCNN-IoU | RT-DETR | |
SECOND | SECOND-IoU | YOLOv9 |
DeepSparkHub 24.06对应天数软件栈3.4.0版本。
感谢以下人员做出的贡献:
majorli,Jino Yang,李一力,shengquan.nian,wenfeng.zhang,吴永乐,zhaojun0044,xiaomei.wang,gaiqin.bai,sanghui-ilu。
欢迎以任何形式为DeepSparkHub项目贡献。
PyTorch | |||||
---|---|---|---|---|---|
BiSeNetV2 | BLIP | CenterPoint | PAConv | PointRCNN | |
PaddlePaddle | |||||
DBNet++ | PP-OCR-EAST | PP-YOLOE+ | SAST | YOLOv5 |
DeepSparkHub 24.03对应天数软件栈3.1.1版本。
感谢以下人员做出的贡献:
majorli,songjian,xiaomei.wang,李一力,gaiqin.bai,wenfeng.zhang,吴永乐,sanghui_ilu,shengquan.nian,yanlong.hao,zhaojun0044,李永康,小豆子。
欢迎以任何形式为DeepSparkHub项目贡献。
PyTorch | |||||
---|---|---|---|---|---|
ACNet | Co-DETR | ControlNet | DBNet++ | DDPM | Efficient Conformer V2 |
Mask2Former | Oriented RepPoints | RepMLP | RepViT | RTMDet | SOLOv2 |
PaddlePaddle | |||||
BlazeFace | ByteTrack | DLRM | DQN | EfficientNetB0 | FastFCN |
FFM | GAT | GraphSAGE | Pix2Pix | PP-HumanSegV1 | Res2Net50_14w_8s |
SE_ResNet50_vd | STDC | PP-TTS-FastSpeech2 | Xception41 | xDeepFM | XLNet |
DeepSparkHub 23.12对应天数软件栈3.1.0版本。
感谢以下人员做出的贡献:
majorli,songjian,may,jun.zhao,wz50525,gaiqin_bai,郝燕龙,巩亚飞,张文风,li.ding,Asltw,吴永乐,westnight,shengquan.nian,la。
欢迎以任何形式为DeepSparkHub项目贡献。
PyTorch | ||||
---|---|---|---|---|
BYOL | InternImage | MobileOne | MoCoV2 | WSLD |
TensorFlow | ||||
AlexNet | DeepLabV3-Plus | FaceNet | InceptionV3 | VGG |
MindSpore | ||||
CRNN | InceptionV3 | MobileNetV3 | OpenPose | ResNeXt50 |
PaddlePaddle | ||||
DenseNet121 | GCN | GPT2-Medium-EN | HRNet-W32 | MobileNetV3_Large1.0 |
MobileSeg | OCRNet | PP-ASR-Conformer | PP-HumanSegV2 | PP-LCNet |
PP-LiteSeg | PP-PicoDet | PP-TTS-FastSpeech2 | PP-TTS-HiFiGAN | ShuffleNetV2 |
DeepSparkHub 23.09对应天数软件栈3.1.0版本。
感谢以下人员做出的贡献:
majorli,zhaojun0044,songjian,gongyafei,gaiqin_bai,Asltw,张文风,吴永乐,丁力,tonychen,牛斯克,la。
欢迎以任何形式为DeepSparkHub项目贡献。
添加了30个基于PyTorch框架的算法模型,新增了网络剪枝、自监督学习、知识蒸馏这3种模型类别。
PyTorch | AlphaPose | ArcFace | Attention U-Net | CBAM | CosFace |
---|---|---|---|---|---|
CspDarknet53 | CWD | DDRNet | FaceNet | FasterNet | |
MAE | Network Slimming | PointNet++ | RKD | STDC | |
UNet++ | YOLOv6 | YOLOv8 | |||
PyTorch (WeNet) | Transformer | U2++ Conformer | Unified Conformer | ||
PyTorch (MMDetection) | ATSS | Cascade R-CNN | CornerNet | DCNV2 | RepPoints |
PyTorch (Fairseq) | BART | Convolutional | RoBERTa | Transformer |
30个模型中有12个使用了开源工具箱,包括:
DeepSparkHub 23.06对应天数软件栈3.0.0版本。
感谢以下人员做出的贡献:
majorli,吴永乐,songjian,丁力,shengquan.nian,may,张文风,chenyingtony,yue.chang,westnight,fhfang,li-shi-kun。
欢迎以任何形式为DeepSparkHub项目贡献。
新增了对TensorFlow和MindSpore的支持。
新增了35个算法模型,具体如下。
TensorFlow | ||||
---|---|---|---|---|
BERT | ResNet50 | SSD | VNet | YOLOv3 |
MindSpore | ||||
BERT | DCGAN | DeepLabV3 | GCN | SSD |
PyTorch | ||||
BERT NER | BERT Question Answering |
BERT Text Classification |
BERT Text Summarization |
BEVFormer |
DBNet | DeepSORT | FCOS | PointPillars | RepVGG |
Retinaface | SOLO | Stable Diffusion | YOLACT++ | YOLOv7 |
PaddlePaddle | ||||
BiSeNetV2 | CenterNet | DeepLabV3+ | DETR | FCOS |
RepVGG | ResNeSt50 | RetinaNet | SOLOv2 | Swin Transformer |
DeepSparkHub 23.03对应天数软件栈3.0.0版本。
感谢以下人员做出的贡献:
shengquan.nian,牛冠博,吴永乐,may,majorli6,yue.chang,wenfeng.zhang,丁力,xiaomei.wang,songjian,guanbo,cheneychen2023,Yili。
欢迎以任何形式为DeepSparkHub项目贡献。
模型名称 | 框架 | 数据集 |
---|---|---|
googlenet | PaddlePaddle | ImageNet |
MobileNetV3 | PaddlePaddle | ImageNet |
ResNet50 | PaddlePaddle | ImageNet |
VGG16 | PaddlePaddle | ImageNet |
Mask R-CNN | PaddlePaddle | COCO |
SSD | PaddlePaddle | COCO |
YOLOv3 | PaddlePaddle | COCO |
PP-YOLOE | PaddlePaddle | COCO |
PSE | PaddlePaddle | OCR_Recog |
CRNN | PaddlePaddle | LMDB |
PP-OCR DB | PaddlePaddle | ICDAR2015 |
DeepLab | PaddlePaddle | COCO |
UNet | PaddlePaddle | CityScapes |
dnlnet | PaddlePaddle | CityScapes |
BERT Pretraining | PaddlePaddle | MNLI |
Ernie | PaddlePaddle | corpus |
Transformer | PaddlePaddle | wmt14-en-de-pre-processed |
DeepFM | PaddlePaddle | Criteo_Terabyte |
Wide&Deep | PaddlePaddle | Criteo_Terabyte |
DeepSparkHub 22.12对应天数软件栈2.3.0版本。
感谢以下人员做出的贡献:
majorli,张文风,yufei.chen,jeff guo,李睿,sanghui_ilu,westnight,jianyong.jiang,
may,Shikun Li,shengquan.nian,qiangzibro,hui.sang。
欢迎以任何形式为DeepSparkHub项目贡献。