From b4d860e1decb1027096d1e84a6b4cd118c7dd767 Mon Sep 17 00:00:00 2001 From: "mingjiang.li" Date: Wed, 18 Dec 2024 17:48:06 +0800 Subject: [PATCH 1/2] minor fix on table format Signed-off-by: mingjiang.li --- README.md | 574 +++++++++++++++++++++++++++--------------------------- 1 file changed, 287 insertions(+), 287 deletions(-) diff --git a/README.md b/README.md index 839860797..944237310 100644 --- a/README.md +++ b/README.md @@ -10,299 +10,299 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[ACmix](cv/classification/acmix/pytorch/README.md) | PyTorch | ImageNet -[ACNet](cv/classification/acnet/pytorch/README.md) | PyTorch | ImageNet -[AlexNet](cv/classification/alexnet/pytorch/README.md) | PyTorch | ImageNet -[AlexNet](cv/classification/alexnet/tensorflow/README.md) | TensorFlow | ImageNet -[BYOL](cv/classification/byol/pytorch/README.md) | PyTorch | ImageNet -[CBAM](cv/classification/cbam/pytorch/README.md)  | PyTorch | ImageNet -[ConvNext](cv/classification/convnext/pytorch/README.md) | PyTorch | ImageNet -[CspDarknet53](cv/classification/cspdarknet53/pytorch/README.md)  | PyTorch | ImageNet -[DenseNet](cv/classification/densenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[DenseNet](cv/classification/densenet/pytorch/README.md) | PyTorch | ImageNet -[DPN92](cv/classification/dpn92/pytorch/README.md) | PyTorch | ImageNet -[DPN107](cv/classification/dpn107/pytorch/README.md) | PyTorch | ImageNet -[ECA-MobileNetV2](cv/classification/eca_mobilenet_v2/pytorch/README.md) | PyTorch | ImageNet -[ECA-ResNet152](cv/classification/eca_resnet152/pytorch/README.md) | PyTorch | ImageNet -[EfficientNetB0](cv/classification/efficientnet_b0/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[EfficientNetB4](cv/classification/efficientnet_b4/pytorch/README.md) | PyTorch | ImageNet -[FasterNet](cv/classification/fasternet/pytorch/README.md)  | PyTorch | ImageNet -[GoogLeNet](cv/classification/googlenet/pytorch/README.md) | PyTorch | ImageNet -[GoogLeNet](cv/classification/googlenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[InceptionV3](cv/classification/inceptionv3/mindspore/README.md) | MindSpore | ImageNet -[InceptionV3](cv/classification/inceptionv3/pytorch/README.md) | PyTorch | ImageNet -[InceptionV3](cv/classification/inceptionv3/tensorflow/README.md) | TensorFlow | ImageNet -[InceptionV4](cv/classification/inceptionv4/pytorch/README.md) | PyTorch | ImageNet -[InternImage](cv/classification/internimage/pytorch/README.md) | PyTorch | ImageNet -[LeNet](cv/classification/lenet/pytorch/README.md) | PyTorch | ImageNet -[MobileNetV2](cv/classification/mobilenetv2/pytorch/README.md) | PyTorch | ImageNet -[MobileNetV3](cv/classification/mobilenetv3/mindspore/README.md) | MindSpore | ImageNet -[MobileNetV3](cv/classification/mobilenetv3/pytorch/README.md) | PyTorch | ImageNet -[MobileNetV3](cv/classification/mobilenetv3/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[ACmix](cv/classification/acmix/pytorch/README.md) | PyTorch | ImageNet +[ACNet](cv/classification/acnet/pytorch/README.md) | PyTorch | ImageNet +[AlexNet](cv/classification/alexnet/pytorch/README.md) | PyTorch | ImageNet +[AlexNet](cv/classification/alexnet/tensorflow/README.md) | TensorFlow | ImageNet +[BYOL](cv/classification/byol/pytorch/README.md) | PyTorch | ImageNet +[CBAM](cv/classification/cbam/pytorch/README.md) | PyTorch | ImageNet +[ConvNext](cv/classification/convnext/pytorch/README.md) | PyTorch | ImageNet +[CspDarknet53](cv/classification/cspdarknet53/pytorch/README.md) | PyTorch | ImageNet +[DenseNet](cv/classification/densenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[DenseNet](cv/classification/densenet/pytorch/README.md) | PyTorch | ImageNet +[DPN92](cv/classification/dpn92/pytorch/README.md) | PyTorch | ImageNet +[DPN107](cv/classification/dpn107/pytorch/README.md) | PyTorch | ImageNet +[ECA-MobileNetV2](cv/classification/eca_mobilenet_v2/pytorch/README.md) | PyTorch | ImageNet +[ECA-ResNet152](cv/classification/eca_resnet152/pytorch/README.md) | PyTorch | ImageNet +[EfficientNetB0](cv/classification/efficientnet_b0/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[EfficientNetB4](cv/classification/efficientnet_b4/pytorch/README.md) | PyTorch | ImageNet +[FasterNet](cv/classification/fasternet/pytorch/README.md) | PyTorch | ImageNet +[GoogLeNet](cv/classification/googlenet/pytorch/README.md) | PyTorch | ImageNet +[GoogLeNet](cv/classification/googlenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[InceptionV3](cv/classification/inceptionv3/mindspore/README.md) | MindSpore | ImageNet +[InceptionV3](cv/classification/inceptionv3/pytorch/README.md) | PyTorch | ImageNet +[InceptionV3](cv/classification/inceptionv3/tensorflow/README.md) | TensorFlow | ImageNet +[InceptionV4](cv/classification/inceptionv4/pytorch/README.md) | PyTorch | ImageNet +[InternImage](cv/classification/internimage/pytorch/README.md) | PyTorch | ImageNet +[LeNet](cv/classification/lenet/pytorch/README.md) | PyTorch | ImageNet +[MobileNetV2](cv/classification/mobilenetv2/pytorch/README.md) | PyTorch | ImageNet +[MobileNetV3](cv/classification/mobilenetv3/mindspore/README.md) | MindSpore | ImageNet +[MobileNetV3](cv/classification/mobilenetv3/pytorch/README.md) | PyTorch | ImageNet +[MobileNetV3](cv/classification/mobilenetv3/paddlepaddle/README.md) | PaddlePaddle | ImageNet [MobileNetV3_Large1.0](cv/classification/mobilenetv3_large_x1_0/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[MobileOne](cv/classification/mobileone/pytorch/README.md) | PyTorch | ImageNet -[MoCoV2](cv/classification/mocov2/pytorch/README.md) | PyTorch | ImageNet -[PP-LCNet](cv/classification/pp-lcnet/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[RepMLP](cv/classification/repmlp/pytorch/README.md) | PyTorch | ImageNet -[RepVGG](cv/classification/repvgg/pytorch/README.md) | PyTorch | ImageNet -[RepVGG](cv/classification/repvgg/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[RepViT](cv/classification/repvit/pytorch/README.md) | PyTorch | ImageNet -[Res2Net50_14w_8s](cv/classification/Res2Net50_14w_8s/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[ResNeSt14](cv/classification/resnest14/pytorch/README.md) | PyTorch | ImageNet -[ResNeSt50](cv/classification/resnest50/pytorch/README.md) | PyTorch | ImageNet -[ResNeSt50](cv/classification/resnest50/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[ResNeSt101](cv/classification/resnest101/pytorch/README.md) | PyTorch | ImageNet -[ResNeSt269](cv/classification/resnest269/pytorch/README.md) | PyTorch | ImageNet -[ResNet18](cv/classification/resnet18/pytorch/README.md) | PyTorch | ImageNet -[ResNet50](cv/classification/resnet50/pytorch/README.md) | PyTorch | ImageNet -[ResNet50](cv/classification/resnet50/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[ResNet50](cv/classification/resnet50/tensorflow/README.md) | TensorFlow | ImageNet -[ResNet101](cv/classification/resnet101/pytorch/README.md) | PyTorch | ImageNet -[ResNet152](cv/classification/resnet152/pytorch/README.md) | PyTorch | ImageNet +[MobileOne](cv/classification/mobileone/pytorch/README.md) | PyTorch | ImageNet +[MoCoV2](cv/classification/mocov2/pytorch/README.md) | PyTorch | ImageNet +[PP-LCNet](cv/classification/pp-lcnet/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[RepMLP](cv/classification/repmlp/pytorch/README.md) | PyTorch | ImageNet +[RepVGG](cv/classification/repvgg/pytorch/README.md) | PyTorch | ImageNet +[RepVGG](cv/classification/repvgg/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[RepViT](cv/classification/repvit/pytorch/README.md) | PyTorch | ImageNet +[Res2Net50_14w_8s](cv/classification/Res2Net50_14w_8s/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[ResNeSt14](cv/classification/resnest14/pytorch/README.md) | PyTorch | ImageNet +[ResNeSt50](cv/classification/resnest50/pytorch/README.md) | PyTorch | ImageNet +[ResNeSt50](cv/classification/resnest50/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[ResNeSt101](cv/classification/resnest101/pytorch/README.md) | PyTorch | ImageNet +[ResNeSt269](cv/classification/resnest269/pytorch/README.md) | PyTorch | ImageNet +[ResNet18](cv/classification/resnet18/pytorch/README.md) | PyTorch | ImageNet +[ResNet50](cv/classification/resnet50/pytorch/README.md) | PyTorch | ImageNet +[ResNet50](cv/classification/resnet50/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[ResNet50](cv/classification/resnet50/tensorflow/README.md) | TensorFlow | ImageNet +[ResNet101](cv/classification/resnet101/pytorch/README.md) | PyTorch | ImageNet +[ResNet152](cv/classification/resnet152/pytorch/README.md) | PyTorch | ImageNet [ResNeXt50_32x4d](cv/classification/resnext50_32x4d/mindspore/README.md) | MindSpore | ImageNet -[ResNeXt50_32x4d](cv/classification/resnext50_32x4d/pytorch/README.md) | PyTorch | ImageNet -[ResNeXt101_32x8d](cv/classification/resnext101_32x8d/pytorch/README.md) | PyTorch | ImageNet -[SE_ResNet50_vd](cv/classification/se_resnet50_vd/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[SEResNeXt](cv/classification/seresnext/pytorch/README.md) | PyTorch | ImageNet -[ShuffleNetV2](cv/classification/shufflenetv2/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[ShuffleNetV2](cv/classification/shufflenetv2/pytorch/README.md) | PyTorch | ImageNet -[SqueezeNet](cv/classification/squeezenet/pytorch/README.md) | PyTorch | ImageNet -[Swin Transformer](cv/classification/swin_transformer/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[Swin Transformer](cv/classification/swin_transformer/pytorch/README.md) | PyTorch | ImageNet -[VGG16](cv/classification/vgg/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[VGG16](cv/classification/vgg/pytorch/README.md) | PyTorch | ImageNet -[VGG16](cv/classification/vgg/tensorflow/README.md) | TensorFlow | ImageNet -[Wave-MLP](cv/classification/wavemlp/pytorch/README.md) | PyTorch | ImageNet -[Wide_ResNet101_2](cv/classification/wide_resnet101_2/pytorch/README.md) | PyTorch | ImageNet -[Xception](cv/classification/xception/paddlepaddle/README.md) | PaddlePaddle | ImageNet -[Xception](cv/classification/xception/pytorch/README.md) | PyTorch | ImageNet +[ResNeXt50_32x4d](cv/classification/resnext50_32x4d/pytorch/README.md) | PyTorch | ImageNet +[ResNeXt101_32x8d](cv/classification/resnext101_32x8d/pytorch/README.md) | PyTorch | ImageNet +[SE_ResNet50_vd](cv/classification/se_resnet50_vd/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[SEResNeXt](cv/classification/seresnext/pytorch/README.md) | PyTorch | ImageNet +[ShuffleNetV2](cv/classification/shufflenetv2/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[ShuffleNetV2](cv/classification/shufflenetv2/pytorch/README.md) | PyTorch | ImageNet +[SqueezeNet](cv/classification/squeezenet/pytorch/README.md) | PyTorch | ImageNet +[Swin Transformer](cv/classification/swin_transformer/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[Swin Transformer](cv/classification/swin_transformer/pytorch/README.md) | PyTorch | ImageNet +[VGG16](cv/classification/vgg/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[VGG16](cv/classification/vgg/pytorch/README.md) | PyTorch | ImageNet +[VGG16](cv/classification/vgg/tensorflow/README.md) | TensorFlow | ImageNet +[Wave-MLP](cv/classification/wavemlp/pytorch/README.md) | PyTorch | ImageNet +[Wide_ResNet101_2](cv/classification/wide_resnet101_2/pytorch/README.md) | PyTorch | ImageNet +[Xception](cv/classification/xception/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[Xception](cv/classification/xception/pytorch/README.md) | PyTorch | ImageNet #### Face Detection 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[RetinaFace](cv/face/retinaface/pytorch/README.md) | PyTorch | WiderFace +[RetinaFace](cv/face/retinaface/pytorch/README.md) | PyTorch | WiderFace #### Face Recognition 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[ArcFace](cv/face/arcface/pytorch/README.md)  | PyTorch | CASIA-WebFaces&LFW -[BlazeFace](cv/face/blazeface/paddlepaddle/README.md)  | PaddlePaddle | WIDER-FACE -[CosFace](cv/face/cosface/pytorch/README.md)  | PyTorch | CASIA-WebFaces&LFW -[FaceNet](cv/face/facenet/pytorch/README.md)  | PyTorch | CASIA-WebFaces&LFW +[ArcFace](cv/face/arcface/pytorch/README.md) | PyTorch | CASIA-WebFaces&LFW +[BlazeFace](cv/face/blazeface/paddlepaddle/README.md) | PaddlePaddle | WIDER-FACE +[CosFace](cv/face/cosface/pytorch/README.md) | PyTorch | CASIA-WebFaces&LFW +[FaceNet](cv/face/facenet/pytorch/README.md) | PyTorch | CASIA-WebFaces&LFW [FaceNet](cv/face/facenet/tensorflow/README.md) | TensorFlow | CASIA-WebFaces&LFW #### Instance Segmentation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[SOLO](cv/instance_segmentation/SOLO/pytorch/README.md) | PyTorch | COCO -[SOLOv2](cv/detection/solov2/paddlepaddle/README.md) | PaddlePaddle | COCO -[SOLOv2](cv/instance_segmentation/solov2/pytorch/README.md) | PyTorch | COCO -[YOLACT++](cv/instance_segmentation/yolact/pytorch/README.md) | PyTorch | COCO +[SOLO](cv/instance_segmentation/SOLO/pytorch/README.md) | PyTorch | COCO +[SOLOv2](cv/detection/solov2/paddlepaddle/README.md) | PaddlePaddle | COCO +[SOLOv2](cv/instance_segmentation/solov2/pytorch/README.md) | PyTorch | COCO +[YOLACT++](cv/instance_segmentation/yolact/pytorch/README.md) | PyTorch | COCO #### Image Generation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[DCGAN](cv/image_generation/dcgan/mindspore/README.md) | MindSpore | ImageNet -[Pix2Pix](cv/image_generation/Pix2pix/paddlepaddle/README.md) | PaddlePaddle | facades +[DCGAN](cv/image_generation/dcgan/mindspore/README.md) | MindSpore | ImageNet +[Pix2Pix](cv/image_generation/Pix2pix/paddlepaddle/README.md) | PaddlePaddle | facades #### Knowledge Distillation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[CWD](cv/distiller/CWD/pytorch/README.md)  | PyTorch | Cityscapes -[RKD](cv/distiller/RKD/pytorch/README.md)  | PyTorch | CUB-200-2011 -[WSLD](cv/distiller/WSLD/pytorch/README.md) | PyTorch | ImageNet +[CWD](cv/distiller/CWD/pytorch/README.md) | PyTorch | Cityscapes +[RKD](cv/distiller/RKD/pytorch/README.md) | PyTorch | CUB-200-2011 +[WSLD](cv/distiller/WSLD/pytorch/README.md) | PyTorch | ImageNet #### Network Pruning 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Network Slimming](cv/Pruning/Network-Slimming/pytorch/README.md)  | PyTorch | CIFAR-10/100 +[Network Slimming](cv/Pruning/Network-Slimming/pytorch/README.md) | PyTorch | CIFAR-10/100 #### Object Detection 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[ATSS](cv/detection/atss_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO -[AutoAssign](cv/detection/autoassign/pytorch/README.md) | PyTorch | COCO -[Cascade R-CNN](cv/detection/cascade_rcnn_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO -[CenterMask2](cv/detection/centermask2/pytorch/README.md) | PyTorch | COCO -[CenterNet](cv/detection/centernet/pytorch/README.md) | PyTorch | COCO -[CenterNet](cv/detection/centernet/paddlepaddle/README.md) | PaddlePaddle | COCO -[Co-DETR](cv/detection/co-detr/pytorch/README.md) | PyTorch | COCO -[CornerNet](cv/detection/cornernet_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO -[DCNV2](cv/detection/dcnv2_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO -[DeepSORT](cv/tracking/deep_sort/pytorch/README.md) | PyTorch | Market-1501 -[DETR](cv/detection/detr/paddlepaddle/README.md) | PaddlePaddle | COCO -[Faster R-CNN](cv/detection/fasterrcnn/pytorch/README.md) | PyTorch | COCO -[FCOS](cv/detection/fcos/paddlepaddle/README.md) | PaddlePaddle | COCO -[FCOS](cv/detection/fcos/pytorch/README.md) | PyTorch | COCO -[Mamba-YOLO](cv/detection/mamba_yolo/pytorch/README.md) | PyTorch | COCO -[Mask R-CNN](cv/detection/maskrcnn/pytorch/README.md) | PyTorch | COCO -[Mask R-CNN](cv/detection/maskrcnn/paddlepaddle/README.md) | PaddlePaddle | COCO -[OC_SORT](cv/detection/oc_sort/paddlepaddle/README.md) | PaddlePaddle | MOT17 -[Oriented RepPoints](cv/detection/oriented_reppoints/pytorch/README.md) | PyTorch | DOTA -[PP-PicoDet](cv/detection/picodet/paddlepaddle/README.md) | PaddlePaddle | COCO -[PP-YOLOE](cv/detection/pp-yoloe/paddlepaddle/README.md) | PaddlePaddle | COCO -[PP-YOLOE+](cv/detection/pp_yoloe+/paddlepaddle/README.md) | PaddlePaddle | COCO -[PVANet](cv/detection/pvanet/pytorch/README.md) | PyTorch | COCO -[RepPoints](cv/detection/reppoints_mmdet/pytorch/README.md)  | PyTorch (MMDetection) | COCO -[RetinaNet](cv/detection/retinanet/pytorch/README.md) | PyTorch | COCO -[RetinaNet](cv/detection/retinanet/paddlepaddle/README.md) | PaddlePaddle | COCO -[RT-DETR](cv/detection/rt-detr/pytorch/README.md) | PyTorch | COCO -[RTMDet](cv/detection/rtmdet/pytorch/README.md) | PyTorch | COCO -[SSD](cv/detection/ssd/pytorch/README.md) | PyTorch | COCO -[SSD](cv/detection/ssd/paddlepaddle/README.md) | PaddlePaddle | COCO -[SSD](cv/detection/ssd/tensorflow/README.md) | TensorFlow | VOC -[SSD](cv/detection/ssd/mindspore/README.md) | MindSpore | COCO -[YOLOF](cv/detection/yolof/pytorch/README.md) | PyTorch | COCO -[YOLOv3](cv/detection/yolov3/pytorch/README.md) | PyTorch | COCO -[YOLOv3](cv/detection/yolov3/paddlepaddle/README.md) | PaddlePaddle | COCO -[YOLOv3](cv/detection/yolov3/tensorflow/README.md) | TensorFlow | VOC -[YOLOv5](cv/detection/yolov5/paddlepaddle/README.md) | PaddlePaddle | COCO -[YOLOv5](cv/detection/yolov5/pytorch/README.md) | PyTorch | COCO -[YOLOv6](cv/detection/yolov6/pytorch/README.md)  | PyTorch | COCO -[YOLOv7](cv/detection/yolov7/pytorch/README.md) | PyTorch | COCO -[YOLOv8](cv/detection/yolov8/pytorch/README.md)  | PyTorch | COCO -[YOLOv9](cv/detection/yolov9/pytorch/README.md)  | PyTorch | COCO -[YOLOv10](cv/detection/yolov10/pytorch/README.md)  | PyTorch | COCO +[ATSS](cv/detection/atss_mmdet/pytorch/README.md) | PyTorch (MMDetection) | COCO +[AutoAssign](cv/detection/autoassign/pytorch/README.md) | PyTorch | COCO +[Cascade R-CNN](cv/detection/cascade_rcnn_mmdet/pytorch/README.md) | PyTorch (MMDetection) | COCO +[CenterMask2](cv/detection/centermask2/pytorch/README.md) | PyTorch | COCO +[CenterNet](cv/detection/centernet/pytorch/README.md) | PyTorch | COCO +[CenterNet](cv/detection/centernet/paddlepaddle/README.md) | PaddlePaddle | COCO +[Co-DETR](cv/detection/co-detr/pytorch/README.md) | PyTorch | COCO +[CornerNet](cv/detection/cornernet_mmdet/pytorch/README.md) | PyTorch (MMDetection) | COCO +[DCNV2](cv/detection/dcnv2_mmdet/pytorch/README.md) | PyTorch (MMDetection) | COCO +[DeepSORT](cv/tracking/deep_sort/pytorch/README.md) | PyTorch | Market-1501 +[DETR](cv/detection/detr/paddlepaddle/README.md) | PaddlePaddle | COCO +[Faster R-CNN](cv/detection/fasterrcnn/pytorch/README.md) | PyTorch | COCO +[FCOS](cv/detection/fcos/paddlepaddle/README.md) | PaddlePaddle | COCO +[FCOS](cv/detection/fcos/pytorch/README.md) | PyTorch | COCO +[Mamba-YOLO](cv/detection/mamba_yolo/pytorch/README.md) | PyTorch | COCO +[Mask R-CNN](cv/detection/maskrcnn/pytorch/README.md) | PyTorch | COCO +[Mask R-CNN](cv/detection/maskrcnn/paddlepaddle/README.md) | PaddlePaddle | COCO +[OC_SORT](cv/detection/oc_sort/paddlepaddle/README.md) | PaddlePaddle | MOT17 +[Oriented RepPoints](cv/detection/oriented_reppoints/pytorch/README.md) | PyTorch | DOTA +[PP-PicoDet](cv/detection/picodet/paddlepaddle/README.md) | PaddlePaddle | COCO +[PP-YOLOE](cv/detection/pp-yoloe/paddlepaddle/README.md) | PaddlePaddle | COCO +[PP-YOLOE+](cv/detection/pp_yoloe+/paddlepaddle/README.md) | PaddlePaddle | COCO +[PVANet](cv/detection/pvanet/pytorch/README.md) | PyTorch | COCO +[RepPoints](cv/detection/reppoints_mmdet/pytorch/README.md) | PyTorch (MMDetection) | COCO +[RetinaNet](cv/detection/retinanet/pytorch/README.md) | PyTorch | COCO +[RetinaNet](cv/detection/retinanet/paddlepaddle/README.md) | PaddlePaddle | COCO +[RT-DETR](cv/detection/rt-detr/pytorch/README.md) | PyTorch | COCO +[RTMDet](cv/detection/rtmdet/pytorch/README.md) | PyTorch | COCO +[SSD](cv/detection/ssd/pytorch/README.md) | PyTorch | COCO +[SSD](cv/detection/ssd/paddlepaddle/README.md) | PaddlePaddle | COCO +[SSD](cv/detection/ssd/tensorflow/README.md) | TensorFlow | VOC +[SSD](cv/detection/ssd/mindspore/README.md) | MindSpore | COCO +[YOLOF](cv/detection/yolof/pytorch/README.md) | PyTorch | COCO +[YOLOv3](cv/detection/yolov3/pytorch/README.md) | PyTorch | COCO +[YOLOv3](cv/detection/yolov3/paddlepaddle/README.md) | PaddlePaddle | COCO +[YOLOv3](cv/detection/yolov3/tensorflow/README.md) | TensorFlow | VOC +[YOLOv5](cv/detection/yolov5/paddlepaddle/README.md) | PaddlePaddle | COCO +[YOLOv5](cv/detection/yolov5/pytorch/README.md) | PyTorch | COCO +[YOLOv6](cv/detection/yolov6/pytorch/README.md) | PyTorch | COCO +[YOLOv7](cv/detection/yolov7/pytorch/README.md) | PyTorch | COCO +[YOLOv8](cv/detection/yolov8/pytorch/README.md) | PyTorch | COCO +[YOLOv9](cv/detection/yolov9/pytorch/README.md) | PyTorch | COCO +[YOLOv10](cv/detection/yolov10/pytorch/README.md) | PyTorch | COCO #### 3D Object Detection 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[BEVFormer](cv/3d_detection/BEVFormer/pytorch/README.md) | PyTorch | nuScenes&CAN bus -[CenterPoint](cv/3d_detection/centerpoint/pytorch/README.md) | PyTorch | nuScenes -[PAConv](cv/3d_detection/PAConv/pytorch/README.md) | PyTorch | S3DIS -[Part-A2-Anchor](cv/3d_detection/part_a2_anchor/pytorch/README.md) | PyTorch | KITTI -[Part-A2-Free](cv/3d_detection/part_a2_free/pytorch/README.md) | PyTorch | KITTI -[PointNet++](cv/3d_detection/pointnet2/pytorch/mmdetection3d/README.md)  | PyTorch | S3DIS -[PointPillars](cv/3d_detection/pointpillars/pytorch/README.md) | PyTorch | KITTI -[PointRCNN](cv/3d_detection/pointrcnn/pytorch/README.md) | PyTorch | KITTI -[PointRCNN-IoU](cv/3d_detection/pointrcnn_iou/pytorch/README.md) | PyTorch | KITTI -[SECOND](cv/3d_detection/second/pytorch/README.md) | PyTorch | KITTI -[SECOND-IoU](cv/3d_detection/second_iou/pytorch/README.md) | PyTorch | KITTI +[BEVFormer](cv/3d_detection/BEVFormer/pytorch/README.md) | PyTorch | nuScenes&CAN bus +[CenterPoint](cv/3d_detection/centerpoint/pytorch/README.md) | PyTorch | nuScenes +[PAConv](cv/3d_detection/PAConv/pytorch/README.md) | PyTorch | S3DIS +[Part-A2-Anchor](cv/3d_detection/part_a2_anchor/pytorch/README.md) | PyTorch | KITTI +[Part-A2-Free](cv/3d_detection/part_a2_free/pytorch/README.md) | PyTorch | KITTI +[PointNet++](cv/3d_detection/pointnet2/pytorch/mmdetection3d/README.md) | PyTorch | S3DIS +[PointPillars](cv/3d_detection/pointpillars/pytorch/README.md) | PyTorch | KITTI +[PointRCNN](cv/3d_detection/pointrcnn/pytorch/README.md) | PyTorch | KITTI +[PointRCNN-IoU](cv/3d_detection/pointrcnn_iou/pytorch/README.md) | PyTorch | KITTI +[SECOND](cv/3d_detection/second/pytorch/README.md) | PyTorch | KITTI +[SECOND-IoU](cv/3d_detection/second_iou/pytorch/README.md) | PyTorch | KITTI #### OCR 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[CRNN](cv/ocr/crnn/mindspore/README.md) | MindSpore | OCR_Recog -[CRNN](cv/ocr/crnn/paddlepaddle/README.md) | PaddlePaddle | LMDB -[DBNet](cv/ocr/dbnet/pytorch/README.md) | PyTorch | ICDAR2015 -[DBNet++](cv/ocr/dbnetpp/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 -[DBNet++](cv/ocr/dbnetpp/pytorch/README.md) | PyTorch | ICDAR2015 -[PP-OCR-DB](cv/ocr/pp-ocr-db/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 -[PP-OCR-EAST](cv/ocr/pp-ocr-east/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 -[PSE](cv/ocr/pse/paddlepaddle/README.md) | PaddlePaddle | OCR_Recog -[SAR](cv/ocr/sar/pytorch/README.md) | PyTorch | OCR_Recog -[SAST](cv/ocr/sast/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 -[SATRN](cv/ocr/satrn/pytorch/base/README.md) | PyTorch | OCR_Recog +[CRNN](cv/ocr/crnn/mindspore/README.md) | MindSpore | OCR_Recog +[CRNN](cv/ocr/crnn/paddlepaddle/README.md) | PaddlePaddle | LMDB +[DBNet](cv/ocr/dbnet/pytorch/README.md) | PyTorch | ICDAR2015 +[DBNet++](cv/ocr/dbnetpp/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 +[DBNet++](cv/ocr/dbnetpp/pytorch/README.md) | PyTorch | ICDAR2015 +[PP-OCR-DB](cv/ocr/pp-ocr-db/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 +[PP-OCR-EAST](cv/ocr/pp-ocr-east/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 +[PSE](cv/ocr/pse/paddlepaddle/README.md) | PaddlePaddle | OCR_Recog +[SAR](cv/ocr/sar/pytorch/README.md) | PyTorch | OCR_Recog +[SAST](cv/ocr/sast/paddlepaddle/README.md) | PaddlePaddle | ICDAR2015 +[SATRN](cv/ocr/satrn/pytorch/base/README.md) | PyTorch | OCR_Recog #### Point Cloud 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Point-BERT](cv/point_cloud/Point-BERT/pytorch/README.md) | PyTorch | ShapeNet55 & processed ModelNet +[Point-BERT](cv/point_cloud/Point-BERT/pytorch/README.md) | PyTorch | ShapeNet55 & processed ModelNet #### Pose Estimation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[AlphaPose](cv/pose/alphapose/pytorch/README.md)  | PyTorch | COCO -[HRNet](cv/pose/hrnet/pytorch/README.md) | PyTorch | COCO -[HRNet-W32](cv/pose/hrnet/paddlepaddle/README.md) | PaddlePaddle | COCO -[OpenPose](cv/pose/openpose/mindspore/README.md) | MindSpore | COCO +[AlphaPose](cv/pose/alphapose/pytorch/README.md) | PyTorch | COCO +[HRNet](cv/pose/hrnet/pytorch/README.md) | PyTorch | COCO +[HRNet-W32](cv/pose/hrnet/paddlepaddle/README.md) | PaddlePaddle | COCO +[OpenPose](cv/pose/openpose/mindspore/README.md) | MindSpore | COCO #### Self-Supervised Learning 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[MAE](cv/self_supervised_learning/MAE/pytorch/README.md)  | PyTorch | ImageNet +[MAE](cv/self_supervised_learning/MAE/pytorch/README.md) | PyTorch | ImageNet #### Semantic Segmentation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[3D-UNet](cv/semantic_segmentation/unet3d/pytorch/README.md) | PyTorch | kits19 -[APCNet](cv/semantic_segmentation/apcnet/pytorch/README.md) | PyTorch | Cityscapes -[Attention U-net](cv/semantic_segmentation/att_unet/pytorch/README.md)  | PyTorch | Cityscapes -[BiSeNet](cv/semantic_segmentation/bisenet/pytorch/README.md) | PyTorch | COCO -[BiSeNetV2](cv/semantic_segmentation/bisenetv2/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[BiSeNetV2](cv/semantic_segmentation/bisenetv2/pytorch/README.md) | PyTorch | Cityscapes -[CGNet](cv/semantic_segmentation/cgnet/pytorch/README.md) | PyTorch | COCO -[ContextNet](cv/semantic_segmentation/contextnet/pytorch/README.md) | PyTorch | COCO -[DabNet](cv/semantic_segmentation/dabnet/pytorch/README.md) | PyTorch | COCO -[DANet](cv/semantic_segmentation/danet/pytorch/README.md) | PyTorch | COCO -[DDRnet](cv/semantic_segmentation/ddrnet/pytorch/README.md)  | PyTorch | Cityscapes -[DeepLabV3](cv/semantic_segmentation/deeplabv3/pytorch/README.md) | PyTorch | COCO -[DeepLabV3](cv/semantic_segmentation/deeplabv3/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[DeepLabV3](cv/semantic_segmentation/deeplabv3/mindspore/README.md) | MindSpore | VOC -[DeepLabV3+](cv/semantic_segmentation/deeplabv3plus/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[3D-UNet](cv/semantic_segmentation/unet3d/pytorch/README.md) | PyTorch | kits19 +[APCNet](cv/semantic_segmentation/apcnet/pytorch/README.md) | PyTorch | Cityscapes +[Attention U-net](cv/semantic_segmentation/att_unet/pytorch/README.md) | PyTorch | Cityscapes +[BiSeNet](cv/semantic_segmentation/bisenet/pytorch/README.md) | PyTorch | COCO +[BiSeNetV2](cv/semantic_segmentation/bisenetv2/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[BiSeNetV2](cv/semantic_segmentation/bisenetv2/pytorch/README.md) | PyTorch | Cityscapes +[CGNet](cv/semantic_segmentation/cgnet/pytorch/README.md) | PyTorch | COCO +[ContextNet](cv/semantic_segmentation/contextnet/pytorch/README.md) | PyTorch | COCO +[DabNet](cv/semantic_segmentation/dabnet/pytorch/README.md) | PyTorch | COCO +[DANet](cv/semantic_segmentation/danet/pytorch/README.md) | PyTorch | COCO +[DDRnet](cv/semantic_segmentation/ddrnet/pytorch/README.md) | PyTorch | Cityscapes +[DeepLabV3](cv/semantic_segmentation/deeplabv3/pytorch/README.md) | PyTorch | COCO +[DeepLabV3](cv/semantic_segmentation/deeplabv3/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[DeepLabV3](cv/semantic_segmentation/deeplabv3/mindspore/README.md) | MindSpore | VOC +[DeepLabV3+](cv/semantic_segmentation/deeplabv3plus/paddlepaddle/README.md) | PaddlePaddle | Cityscapes [DeepLabV3+](cv/semantic_segmentation/deeplabv3plus/tensorflow/README.md) | TensorFlow | Cityscapes -[DenseASPP](cv/semantic_segmentation/denseaspp/pytorch/README.md) | PyTorch | COCO -[DFANet](cv/semantic_segmentation/dfanet/pytorch/README.md) | PyTorch | COCO -[DNLNet](cv/semantic_segmentation/dnlnet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[DUNet](cv/semantic_segmentation/dunet/pytorch/README.md) | PyTorch | COCO -[EncNet](cv/semantic_segmentation/encnet/pytorch/README.md) | PyTorch | COCO -[ENet](cv/semantic_segmentation/enet/pytorch/README.md) | PyTorch | COCO -[ERFNet](cv/semantic_segmentation/erfnet/pytorch/README.md) | PyTorch | COCO -[ESPNet](cv/semantic_segmentation/espnet/pytorch/README.md) | PyTorch | COCO -[FastFCN](cv/semantic_segmentation/fastfcn/paddlepaddle/README.md) | PyTorch | ADE20K -[FastSCNN](cv/semantic_segmentation/fastscnn/pytorch/README.md) | PyTorch | COCO -[FCN](cv/semantic_segmentation/fcn/pytorch/README.md) | PyTorch | COCO -[FPENet](cv/semantic_segmentation/fpenet/pytorch/README.md) | PyTorch | COCO -[GCNet](cv/semantic_segmentation/gcnet/pytorch/README.md) | PyTorch | Cityscapes -[HardNet](cv/semantic_segmentation/hardnet/pytorch/README.md) | PyTorch | COCO -[ICNet](cv/semantic_segmentation/icnet/pytorch/README.md) | PyTorch | COCO -[LedNet](cv/semantic_segmentation/lednet/pytorch/README.md) | PyTorch | COCO -[LinkNet](cv/semantic_segmentation/linknet/pytorch/README.md) | PyTorch | COCO -[Mask2Former](cv/semantic_segmentation/Mask2Former/pytorch/README.md) | PyTorch | Cityscapes -[MobileSeg](cv/semantic_segmentation/mobileseg/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[OCNet](cv/semantic_segmentation/ocnet/pytorch/README.md) | PyTorch | COCO -[OCRNet](cv/semantic_segmentation/ocrnet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[OCRNet](cv/semantic_segmentation/ocrnet/pytorch/README.md) | PyTorch | Cityscapes -[PP-HumanSegV1](cv/semantic_segmentation/pp_humansegv1/paddlepaddle/README.md) | PaddlePaddle | PP-HumanSeg14K -[PP-HumanSegV2](cv/semantic_segmentation/pp_humansegv2/paddlepaddle/README.md) | PaddlePaddle | PP-HumanSeg14K -[PP-LiteSeg](cv/semantic_segmentation/pp_liteseg/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[PSANet](cv/semantic_segmentation/psanet/pytorch/README.md) | PyTorch | COCO -[RefineNet](cv/semantic_segmentation/refinenet/pytorch/README.md) | PyTorch | COCO -[SegNet](cv/semantic_segmentation/segnet/pytorch/README.md) | PyTorch | COCO -[STDC](cv/semantic_segmentation/stdc/paddlepaddle/README.md)  | PaddlePaddle | Cityscapes -[STDC](cv/semantic_segmentation/stdc/pytorch/README.md)  | PyTorch | Cityscapes -[UNet](cv/semantic_segmentation/unet/pytorch/README.md) | PyTorch | COCO -[UNet](cv/semantic_segmentation/unet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes -[UNet++](cv/semantic_segmentation/unet++/pytorch/README.md)  | PyTorch | DRIVE -[VNet](cv/semantic_segmentation/vnet/tensorflow/README.md) | TensorFlow | Hippocampus +[DenseASPP](cv/semantic_segmentation/denseaspp/pytorch/README.md) | PyTorch | COCO +[DFANet](cv/semantic_segmentation/dfanet/pytorch/README.md) | PyTorch | COCO +[DNLNet](cv/semantic_segmentation/dnlnet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[DUNet](cv/semantic_segmentation/dunet/pytorch/README.md) | PyTorch | COCO +[EncNet](cv/semantic_segmentation/encnet/pytorch/README.md) | PyTorch | COCO +[ENet](cv/semantic_segmentation/enet/pytorch/README.md) | PyTorch | COCO +[ERFNet](cv/semantic_segmentation/erfnet/pytorch/README.md) | PyTorch | COCO +[ESPNet](cv/semantic_segmentation/espnet/pytorch/README.md) | PyTorch | COCO +[FastFCN](cv/semantic_segmentation/fastfcn/paddlepaddle/README.md) | PyTorch | ADE20K +[FastSCNN](cv/semantic_segmentation/fastscnn/pytorch/README.md) | PyTorch | COCO +[FCN](cv/semantic_segmentation/fcn/pytorch/README.md) | PyTorch | COCO +[FPENet](cv/semantic_segmentation/fpenet/pytorch/README.md) | PyTorch | COCO +[GCNet](cv/semantic_segmentation/gcnet/pytorch/README.md) | PyTorch | Cityscapes +[HardNet](cv/semantic_segmentation/hardnet/pytorch/README.md) | PyTorch | COCO +[ICNet](cv/semantic_segmentation/icnet/pytorch/README.md) | PyTorch | COCO +[LedNet](cv/semantic_segmentation/lednet/pytorch/README.md) | PyTorch | COCO +[LinkNet](cv/semantic_segmentation/linknet/pytorch/README.md) | PyTorch | COCO +[Mask2Former](cv/semantic_segmentation/Mask2Former/pytorch/README.md) | PyTorch | Cityscapes +[MobileSeg](cv/semantic_segmentation/mobileseg/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[OCNet](cv/semantic_segmentation/ocnet/pytorch/README.md) | PyTorch | COCO +[OCRNet](cv/semantic_segmentation/ocrnet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[OCRNet](cv/semantic_segmentation/ocrnet/pytorch/README.md) | PyTorch | Cityscapes +[PP-HumanSegV1](cv/semantic_segmentation/pp_humansegv1/paddlepaddle/README.md) | PaddlePaddle | PP-HumanSeg14K +[PP-HumanSegV2](cv/semantic_segmentation/pp_humansegv2/paddlepaddle/README.md) | PaddlePaddle | PP-HumanSeg14K +[PP-LiteSeg](cv/semantic_segmentation/pp_liteseg/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[PSANet](cv/semantic_segmentation/psanet/pytorch/README.md) | PyTorch | COCO +[RefineNet](cv/semantic_segmentation/refinenet/pytorch/README.md) | PyTorch | COCO +[SegNet](cv/semantic_segmentation/segnet/pytorch/README.md) | PyTorch | COCO +[STDC](cv/semantic_segmentation/stdc/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[STDC](cv/semantic_segmentation/stdc/pytorch/README.md) | PyTorch | Cityscapes +[UNet](cv/semantic_segmentation/unet/pytorch/README.md) | PyTorch | COCO +[UNet](cv/semantic_segmentation/unet/paddlepaddle/README.md) | PaddlePaddle | Cityscapes +[UNet++](cv/semantic_segmentation/unet++/pytorch/README.md) | PyTorch | DRIVE +[VNet](cv/semantic_segmentation/vnet/tensorflow/README.md) | TensorFlow | Hippocampus #### Super Resolution 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[basicVSR++](cv/super_resolution/basicVSR++/pytorch/README.md) | PyTorch | REDS -[basicVSR](cv/super_resolution/basicVSR/pytorch/README.md) | PyTorch | REDS -[ESRGAN](cv/super_resolution/esrgan/pytorch/README.md) | PyTorch | DIV2K -[LIIF](cv/super_resolution/liif/pytorch/README.md) | PyTorch | DIV2K -[RealBasicVSR](cv/super_resolution/real_basicVSR/pytorch/README.md) | PyTorch | REDS -[TTSR](cv/super_resolution/ttsr/pytorch/README.md) | PyTorch | CUFED -[TTVSR](cv/super_resolution/ttvsr/pytorch/README.md) | PyTorch | REDS +[basicVSR++](cv/super_resolution/basicVSR++/pytorch/README.md) | PyTorch | REDS +[basicVSR](cv/super_resolution/basicVSR/pytorch/README.md) | PyTorch | REDS +[ESRGAN](cv/super_resolution/esrgan/pytorch/README.md) | PyTorch | DIV2K +[LIIF](cv/super_resolution/liif/pytorch/README.md) | PyTorch | DIV2K +[RealBasicVSR](cv/super_resolution/real_basicVSR/pytorch/README.md) | PyTorch | REDS +[TTSR](cv/super_resolution/ttsr/pytorch/README.md) | PyTorch | CUFED +[TTVSR](cv/super_resolution/ttvsr/pytorch/README.md) | PyTorch | REDS #### Tracking 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[ByteTrack](cv/tracking/bytetrack/paddlepaddle/README.md) | PaddlePaddle | MOT17 -[FairMOT](cv/tracking/fairmot/pytorch/README.md) | PyTorch | MOT17 +[ByteTrack](cv/tracking/bytetrack/paddlepaddle/README.md) | PaddlePaddle | MOT17 +[FairMOT](cv/tracking/fairmot/pytorch/README.md) | PyTorch | MOT17 #### Traffic Forecast 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Graph WaveNet](cv/traffic_forecast/graph_wavenet/pytorch/README.md) | PyTorch | METR-LA & PEMS-BAY +[Graph WaveNet](cv/traffic_forecast/graph_wavenet/pytorch/README.md) | PyTorch | METR-LA & PEMS-BAY ### GNN @@ -310,7 +310,7 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[GAT](gnn/graph_attention/gat/paddlepaddle/README.md) | PaddlePaddle | CORA +[GAT](gnn/graph_attention/gat/paddlepaddle/README.md) | PaddlePaddle | CORA #### Node Classification @@ -322,7 +322,7 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[GCN](gnn/text_classification/GCN/mindspore/README.md) | MindSpore | CORA & Citeseer +[GCN](gnn/text_classification/GCN/mindspore/README.md) | MindSpore | CORA & Citeseer [GCN](gnn/text_classification/GCN/paddlepaddle/README.md) | PaddlePaddle | CORA & PubMed & Citeseer ### HPC @@ -331,31 +331,31 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Water/se_e2_a](hpc/molecular_dynamics/water_se_e2_a/tensorflow/README.md) | TensorFlow (DeePMD-kit) | data_water +[Water/se_e2_a](hpc/molecular_dynamics/water_se_e2_a/tensorflow/README.md) | TensorFlow (DeePMD-kit) | data_water -### Methodology +### Methodology #### Kolmogorov-Arnold Networks 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[KAN](methodology/kolmogorov_arnold_networks/kan/pytorch/README.md) | PyTorch | - +[KAN](methodology/kolmogorov_arnold_networks/kan/pytorch/README.md) | PyTorch | - ### Multimodal 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[BLIP](multimodal/BLIP/pytorch/README.md) | PyTorch | COCO -[CLIP](multimodal/Language-Image_Pre-Training/clip/pytorch/README.md) | PyTorch | CIFAR100 -[ControlNet](multimodal/diffusion/ControlNet/README.md) | PyTorch | Fill50K -[DDPM](multimodal/diffusion/ddpm/README.md) | PyTorch | CIFAR-10 -[LLaVA 1.5](multimodal/llava/pytorch/README.md) | PyTorch | LLaVA-Pretrain -[L-Verse](multimodal/Language-Image_Pre-Training/L-Verse/pytorch/README.md) | PyTorch | ImageNet -[Stable Diffusion 1.4](multimodal/diffusion/stable-diffusion/training/README.md) | PyTorch | pokemon-images -[Stable Diffusion 1.5](multimodal/diffusion/stable-diffusion/sd_1.5/README.md) | PyTorch | pokemon-images -[Stable Diffusion 2.1](multimodal/diffusion/stable-diffusion/sd_2.1/README.md) | PyTorch | pokemon-images -[Stable Diffusion 3](multimodal/diffusion/stable-diffusion/sd_3/README.md) | PyTorch | dog-example -[Stable Diffusion XL](multimodal/diffusion/stable-diffusion/sd_xl/README.md) | PyTorch | pokemon-images +[BLIP](multimodal/BLIP/pytorch/README.md) | PyTorch | COCO +[CLIP](multimodal/Language-Image_Pre-Training/clip/pytorch/README.md) | PyTorch | CIFAR100 +[ControlNet](multimodal/diffusion/ControlNet/README.md) | PyTorch | Fill50K +[DDPM](multimodal/diffusion/ddpm/README.md) | PyTorch | CIFAR-10 +[LLaVA 1.5](multimodal/llava/pytorch/README.md) | PyTorch | LLaVA-Pretrain +[L-Verse](multimodal/Language-Image_Pre-Training/L-Verse/pytorch/README.md) | PyTorch | ImageNet +[Stable Diffusion 1.4](multimodal/diffusion/stable-diffusion/training/README.md) | PyTorch | pokemon-images +[Stable Diffusion 1.5](multimodal/diffusion/stable-diffusion/sd_1.5/README.md) | PyTorch | pokemon-images +[Stable Diffusion 2.1](multimodal/diffusion/stable-diffusion/sd_2.1/README.md) | PyTorch | pokemon-images +[Stable Diffusion 3](multimodal/diffusion/stable-diffusion/sd_3/README.md) | PyTorch | dog-example +[Stable Diffusion XL](multimodal/diffusion/stable-diffusion/sd_xl/README.md) | PyTorch | pokemon-images ### NLP @@ -363,72 +363,72 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[GLM](nlp/cloze_test/glm/pytorch/GLMForMultiTokenCloze/README.md) | PyTorch | GLMForMultiTokenCloze +[GLM](nlp/cloze_test/glm/pytorch/GLMForMultiTokenCloze/README.md) | PyTorch | GLMForMultiTokenCloze #### Dialogue Generation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[CPM](nlp/dialogue_generation/cpm/pytorch/README.md) | PyTorch | STC +[CPM](nlp/dialogue_generation/cpm/pytorch/README.md) | PyTorch | STC #### Language Modeling 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[BART](nlp/language_model/bart_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | RTE -[BERT NER](nlp/ner/bert/pytorch/README.md) | PyTorch | CoNLL-2003 -[BERT Pretraining](nlp/language_model/bert/pytorch/README.md) | PyTorch | MLCommon Wikipedia (2048_shards_uncompressed) -[BERT Pretraining](nlp/language_model/bert/paddlepaddle/README.md) | PaddlePaddle | MNLI -[BERT Pretraining](nlp/language_model/bert/tensorflow/base/README.md) | TensorFlow | MNLI -[BERT Pretraining](nlp/language_model/bert/mindspore/README.md) | MindSpore | SQuAD -[BERT Text Classification](nlp/text_classification/bert/pytorch/README.md) |PyTorch | GLUE -[BERT Text Summerization](nlp/text_summarisation/bert/pytorch/README.md) | PyTorch | cnn_dailymail -[BERT Question Answering](nlp/question_answering/bert/pytorch/README.md) | PyTorch | SQuAD -[GPT2-Medium-EN](nlp/llm/GPT/paddlepaddle/README.md) | PaddlePaddle | SST-2 -[RoBERTa](nlp/language_model/roberta_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | RTE -[XLNet](nlp/language_model/xlnet/paddlepaddle/README.md)  | PaddlePaddle | SST-2 +[BART](nlp/language_model/bart_fairseq/pytorch/README.md) | PyTorch (Fairseq) | RTE +[BERT NER](nlp/ner/bert/pytorch/README.md) | PyTorch | CoNLL-2003 +[BERT Pretraining](nlp/language_model/bert/pytorch/README.md) | PyTorch | MLCommon Wikipedia (2048_shards_uncompressed) +[BERT Pretraining](nlp/language_model/bert/paddlepaddle/README.md) | PaddlePaddle | MNLI +[BERT Pretraining](nlp/language_model/bert/tensorflow/base/README.md) | TensorFlow | MNLI +[BERT Pretraining](nlp/language_model/bert/mindspore/README.md) | MindSpore | SQuAD +[BERT Text Classification](nlp/text_classification/bert/pytorch/README.md) |PyTorch | GLUE +[BERT Text Summerization](nlp/text_summarisation/bert/pytorch/README.md) | PyTorch | cnn_dailymail +[BERT Question Answering](nlp/question_answering/bert/pytorch/README.md) | PyTorch | SQuAD +[GPT2-Medium-EN](nlp/llm/GPT/paddlepaddle/README.md) | PaddlePaddle | SST-2 +[RoBERTa](nlp/language_model/roberta_fairseq/pytorch/README.md) | PyTorch (Fairseq) | RTE +[XLNet](nlp/language_model/xlnet/paddlepaddle/README.md) | PaddlePaddle | SST-2 #### Large Language Model (LLM) 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Aquila2-34B](nlp/llm/aquila2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Baichuan2-7B](nlp/llm/baichuan2-7b/Baichuan2/README.md) | PyTorch (DeepSpeed) | baichuan2-7b-base -[Bloom-7B1](nlp/llm/bloom-7b1/firefly/README.md) | PyTorch (Firefly) | school_math_0.25M & bloom-7b1 -[ChatGLM-6B](nlp/llm/chatglm-6b/deepspeed/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm-6b -[ChatGLM2-6B SFT](nlp/llm/ChatGLM2-6b-sft/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm2-6b -[ChatGLM3-6B](nlp/llm/chatglm3-6b/deepspeed/finetune_demo/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm3-6b -[DeepSeekMoE 7B](nlp/llm/deepseek_moe_7b/colossalai/README.md) | PyTorch (ColossalAI) | deepseek-moe-16b-base -[Llama-7B](nlp/llm/llama-7b/colossalai/README.md) | PyTorch (Colossal-AI) | llama-7b-hf -[Llama2-7B](nlp/llm/llama2-7b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama2-7B Reward Model Finetuning](nlp/llm/llama2-7b_reward_sft/deepspeed/README.md) | PyTorch (DeepSpeed) | Dahoas/rm-static -[Llama2-7B RLHF](nlp/llm/llama2-7b_rlhf/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | llama2-7b&tiny-llama -[Llama2-7B SFT](nlp/llm/llama2-7b_sft/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | GPT Small-117M -[Llama2-13B](nlp/llm/llama2-13b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama2-34B](nlp/llm/llama2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama3-8B](nlp/llm/llama3_8b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama3-8B SFT](nlp/llm/llama3_8b/ColossalAI/applications/Colossal-LLaMA/README.md) | PyTorch (ColossalAI) | school_math_0.25M -[Mamba-2](nlp/llm/mamba-2/megatron-lm/README.md) | PyTorch (Megatron-LM) | GPT Small-117M -[Mixtral 8x7B](nlp/llm/mixtral/megatron-lm/README.md) | PyTorch (Megatron-LM) | GPT Small-117M -[QWen-7B](nlp/llm/qwen-7b/firefly/README.md) | PyTorch (Firefly) | qwen-7b -[QWen1.5-7B](nlp/llm/qwen1.5-7b/firefly/README.md) | PyTorch (Firefly) | school_math -[QWen1.5-14B](nlp/llm/qwen1.5-14b/firefly/README.md) | PyTorch (Firefly) | school_math -[Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/LLaMA-Factory/README.md) | PyTorch (LLaMA-Factory) | qwen2.5-7b +[Aquila2-34B](nlp/llm/aquila2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus +[Baichuan2-7B](nlp/llm/baichuan2-7b/Baichuan2/README.md) | PyTorch (DeepSpeed) | baichuan2-7b-base +[Bloom-7B1](nlp/llm/bloom-7b1/firefly/README.md) | PyTorch (Firefly) | school_math_0.25M & bloom-7b1 +[ChatGLM-6B](nlp/llm/chatglm-6b/deepspeed/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm-6b +[ChatGLM2-6B SFT](nlp/llm/ChatGLM2-6b-sft/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm2-6b +[ChatGLM3-6B](nlp/llm/chatglm3-6b/deepspeed/finetune_demo/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm3-6b +[DeepSeekMoE 7B](nlp/llm/deepseek_moe_7b/colossalai/README.md) | PyTorch (ColossalAI) | deepseek-moe-16b-base +[Llama-7B](nlp/llm/llama-7b/colossalai/README.md) | PyTorch (Colossal-AI) | llama-7b-hf +[Llama2-7B](nlp/llm/llama2-7b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus +[Llama2-7B Reward Model Finetuning](nlp/llm/llama2-7b_reward_sft/deepspeed/README.md) | PyTorch (DeepSpeed) | Dahoas/rm-static +[Llama2-7B RLHF](nlp/llm/llama2-7b_rlhf/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | llama2-7b&tiny-llama +[Llama2-7B SFT](nlp/llm/llama2-7b_sft/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | GPT Small-117M +[Llama2-13B](nlp/llm/llama2-13b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus +[Llama2-34B](nlp/llm/llama2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus +[Llama3-8B](nlp/llm/llama3_8b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus +[Llama3-8B SFT](nlp/llm/llama3_8b/ColossalAI/applications/Colossal-LLaMA/README.md) | PyTorch (ColossalAI) | school_math_0.25M +[Mamba-2](nlp/llm/mamba-2/megatron-lm/README.md) | PyTorch (Megatron-LM) | GPT Small-117M +[Mixtral 8x7B](nlp/llm/mixtral/megatron-lm/README.md) | PyTorch (Megatron-LM) | GPT Small-117M +[QWen-7B](nlp/llm/qwen-7b/firefly/README.md) | PyTorch (Firefly) | qwen-7b +[QWen1.5-7B](nlp/llm/qwen1.5-7b/firefly/README.md) | PyTorch (Firefly) | school_math +[QWen1.5-14B](nlp/llm/qwen1.5-14b/firefly/README.md) | PyTorch (Firefly) | school_math +[Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/LLaMA-Factory/README.md) | PyTorch (LLaMA-Factory) | qwen2.5-7b #### Text Correction 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Ernie](nlp/text_correction/ernie/paddlepaddle/README.md) | PaddlePaddle | corpus +[Ernie](nlp/text_correction/ernie/paddlepaddle/README.md) | PaddlePaddle | corpus #### Translation 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Convolutional](nlp/translation/convolutional_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | WMT14 -[T5](nlp/translation/t5/pytorch/README.md) | PyTorch | wmt14-en-de-pre-processed -[Transformer](nlp/translation/transformer/paddlepaddle/README.md) | PaddlePaddle | wmt14-en-de-pre-processed -[Transformer](nlp/translation/transformer_fairseq/pytorch/README.md)  | PyTorch (Fairseq) | IWSLT14 +[Convolutional](nlp/translation/convolutional_fairseq/pytorch/README.md) | PyTorch (Fairseq) | WMT14 +[T5](nlp/translation/t5/pytorch/README.md) | PyTorch | wmt14-en-de-pre-processed +[Transformer](nlp/translation/transformer/paddlepaddle/README.md) | PaddlePaddle | wmt14-en-de-pre-processed +[Transformer](nlp/translation/transformer_fairseq/pytorch/README.md) | PyTorch (Fairseq) | IWSLT14 ### Recommendation @@ -436,24 +436,24 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[NCF](recommendation/collaborative_filtering/ncf/pytorch/README.md) | PyTorch | movielens +[NCF](recommendation/collaborative_filtering/ncf/pytorch/README.md) | PyTorch | movielens #### Click Through Rate 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[DLRM](recommendation/ctr/dlrm/pytorch/README.md) | PyTorch | Criteo_Terabyte -[DLRM](recommendation/ctr/dlrm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte -[FFM](recommendation/ctr/ffm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte -[DeepFM](recommendation/ctr/deepfm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte -[Wide&Deep](recommendation/ctr/wide_deep/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte -[xDeepFM](recommendation/ctr/xdeepfm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte +[DLRM](recommendation/ctr/dlrm/pytorch/README.md) | PyTorch | Criteo_Terabyte +[DLRM](recommendation/ctr/dlrm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte +[FFM](recommendation/ctr/ffm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte +[DeepFM](recommendation/ctr/deepfm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte +[Wide&Deep](recommendation/ctr/wide_deep/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte +[xDeepFM](recommendation/ctr/xdeepfm/paddlepaddle/README.md) | PaddlePaddle | Criteo_Terabyte ### Reinforcement Learning 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[DQN](reinforcement_learning/DQN/paddlepaddle/README.md) | PaddlePaddle | CartPole-v0 +[DQN](reinforcement_learning/DQN/paddlepaddle/README.md) | PaddlePaddle | CartPole-v0 ### Speech @@ -461,29 +461,29 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[Conformer](speech/speech_recognition/conformer_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL -[Efficient Conformer v2](speech/speech_recognition/efficient_conformer_v2_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL -[PP-ASR-Conformer](speech/speech_recognition/conformer/paddlepaddle/README.md) | PaddlePaddle | AISHELL -[RNN-T](speech/speech_recognition/rnnt/pytorch/README.md) | PyTorch | LJSpeech -[Transformer](speech/speech_recognition/transformer_wenet/pytorch/README.md)  | PyTorch (WeNet) | AISHELL -[U2++ Conformer](speech/speech_recognition/u2++_conformer_wenet/pytorch/README.md)  | PyTorch (WeNet) | AISHELL -[Unified Conformer](speech/speech_recognition/unified_conformer_wenet/pytorch/README.md)  | PyTorch (WeNet) | AISHELL +[Conformer](speech/speech_recognition/conformer_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL +[Efficient Conformer v2](speech/speech_recognition/efficient_conformer_v2_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL +[PP-ASR-Conformer](speech/speech_recognition/conformer/paddlepaddle/README.md) | PaddlePaddle | AISHELL +[RNN-T](speech/speech_recognition/rnnt/pytorch/README.md) | PyTorch | LJSpeech +[Transformer](speech/speech_recognition/transformer_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL +[U2++ Conformer](speech/speech_recognition/u2++_conformer_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL +[Unified Conformer](speech/speech_recognition/unified_conformer_wenet/pytorch/README.md) | PyTorch (WeNet) | AISHELL #### Speech Synthesis 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[PP-TTS-FastSpeech2](speech/speech_synthesis/fastspeech2/paddlepaddle/README.md) | PaddlePaddle | CSMSC -[PP-TTS-HiFiGAN](speech/speech_synthesis/hifigan/paddlepaddle/README.md) | PaddlePaddle | CSMSC -[Tacotron2](speech/speech_synthesis/tacotron2/pytorch/README.md) | PyTorch | LJSpeech -[VQMIVC](speech/speech_synthesis/vqmivc/pytorch/README.md) | PyTorch | VCTK-Corpus -[WaveGlow](speech/speech_synthesis/waveglow/pytorch/README.md) | PyTorch | LJSpeech +[PP-TTS-FastSpeech2](speech/speech_synthesis/fastspeech2/paddlepaddle/README.md) | PaddlePaddle | CSMSC +[PP-TTS-HiFiGAN](speech/speech_synthesis/hifigan/paddlepaddle/README.md) | PaddlePaddle | CSMSC +[Tacotron2](speech/speech_synthesis/tacotron2/pytorch/README.md) | PyTorch | LJSpeech +[VQMIVC](speech/speech_synthesis/vqmivc/pytorch/README.md) | PyTorch | VCTK-Corpus +[WaveGlow](speech/speech_synthesis/waveglow/pytorch/README.md) | PyTorch | LJSpeech ### 3D Reconstruction 模型名称 | 框架 | 数据集 -------- | ------ | ---- -[HashNeRF](3d-reconstruction/ngp-nerf/pytorch/README.md) | PyTorch | fox +[HashNeRF](3d-reconstruction/ngp-nerf/pytorch/README.md) | PyTorch | fox -------- -- Gitee From 2de04afafffe0877e26c6ea30fbf7978873b7d5f Mon Sep 17 00:00:00 2001 From: "mingjiang.li" Date: Wed, 18 Dec 2024 18:05:12 +0800 Subject: [PATCH 2/2] add toolbox table head for llm Signed-off-by: mingjiang.li --- README.md | 48 ++++++++++++++++++++++++------------------------ 1 file changed, 24 insertions(+), 24 deletions(-) diff --git a/README.md b/README.md index 944237310..ea8d38208 100644 --- a/README.md +++ b/README.md @@ -390,30 +390,30 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 #### Large Language Model (LLM) -模型名称 | 框架 | 数据集 --------- | ------ | ---- -[Aquila2-34B](nlp/llm/aquila2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Baichuan2-7B](nlp/llm/baichuan2-7b/Baichuan2/README.md) | PyTorch (DeepSpeed) | baichuan2-7b-base -[Bloom-7B1](nlp/llm/bloom-7b1/firefly/README.md) | PyTorch (Firefly) | school_math_0.25M & bloom-7b1 -[ChatGLM-6B](nlp/llm/chatglm-6b/deepspeed/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm-6b -[ChatGLM2-6B SFT](nlp/llm/ChatGLM2-6b-sft/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm2-6b -[ChatGLM3-6B](nlp/llm/chatglm3-6b/deepspeed/finetune_demo/README.md) | PyTorch (DeepSpeed) | ADGEN & chatglm3-6b -[DeepSeekMoE 7B](nlp/llm/deepseek_moe_7b/colossalai/README.md) | PyTorch (ColossalAI) | deepseek-moe-16b-base -[Llama-7B](nlp/llm/llama-7b/colossalai/README.md) | PyTorch (Colossal-AI) | llama-7b-hf -[Llama2-7B](nlp/llm/llama2-7b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama2-7B Reward Model Finetuning](nlp/llm/llama2-7b_reward_sft/deepspeed/README.md) | PyTorch (DeepSpeed) | Dahoas/rm-static -[Llama2-7B RLHF](nlp/llm/llama2-7b_rlhf/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | llama2-7b&tiny-llama -[Llama2-7B SFT](nlp/llm/llama2-7b_sft/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | GPT Small-117M -[Llama2-13B](nlp/llm/llama2-13b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama2-34B](nlp/llm/llama2-34b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama3-8B](nlp/llm/llama3_8b/megatron-deepspeed/README.md) | PyTorch (Megatron-DeepSpeed) | Bookcorpus -[Llama3-8B SFT](nlp/llm/llama3_8b/ColossalAI/applications/Colossal-LLaMA/README.md) | PyTorch (ColossalAI) | school_math_0.25M -[Mamba-2](nlp/llm/mamba-2/megatron-lm/README.md) | PyTorch (Megatron-LM) | GPT Small-117M -[Mixtral 8x7B](nlp/llm/mixtral/megatron-lm/README.md) | PyTorch (Megatron-LM) | GPT Small-117M -[QWen-7B](nlp/llm/qwen-7b/firefly/README.md) | PyTorch (Firefly) | qwen-7b -[QWen1.5-7B](nlp/llm/qwen1.5-7b/firefly/README.md) | PyTorch (Firefly) | school_math -[QWen1.5-14B](nlp/llm/qwen1.5-14b/firefly/README.md) | PyTorch (Firefly) | school_math -[Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/LLaMA-Factory/README.md) | PyTorch (LLaMA-Factory) | qwen2.5-7b +模型名称 | 框架 | 工具箱 | 数据集/权重 +------- | --- | ----- | ----- +[Aquila2-34B](nlp/llm/aquila2-34b/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | Bookcorpus +[Baichuan2-7B](nlp/llm/baichuan2-7b/Baichuan2/README.md) | PyTorch | DeepSpeed | baichuan2-7b-base +[Bloom-7B1](nlp/llm/bloom-7b1/firefly/README.md) | PyTorch | Firefly | school_math_0.25M & bloom-7b1 +[ChatGLM-6B](nlp/llm/chatglm-6b/deepspeed/README.md) | PyTorch | DeepSpeed | ADGEN & chatglm-6b +[ChatGLM2-6B SFT](nlp/llm/ChatGLM2-6b-sft/README.md) | PyTorch | DeepSpeed | ADGEN & chatglm2-6b +[ChatGLM3-6B](nlp/llm/chatglm3-6b/deepspeed/finetune_demo/README.md) | PyTorch | DeepSpeed | ADGEN & chatglm3-6b +[DeepSeekMoE 7B](nlp/llm/deepseek_moe_7b/colossalai/README.md) | PyTorch | ColossalAI | deepseek-moe-16b-base +[Llama-7B](nlp/llm/llama-7b/colossalai/README.md) | PyTorch | Colossal-AI | llama-7b-hf +[Llama2-7B](nlp/llm/llama2-7b/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | Bookcorpus +[Llama2-7B Reward Model Finetuning](nlp/llm/llama2-7b_reward_sft/deepspeed/README.md) | PyTorch | DeepSpeed | Dahoas/rm-static +[Llama2-7B RLHF](nlp/llm/llama2-7b_rlhf/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | llama2-7b&tiny-llama +[Llama2-7B SFT](nlp/llm/llama2-7b_sft/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | GPT Small-117M +[Llama2-13B](nlp/llm/llama2-13b/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | Bookcorpus +[Llama2-34B](nlp/llm/llama2-34b/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | Bookcorpus +[Llama3-8B](nlp/llm/llama3_8b/megatron-deepspeed/README.md) | PyTorch | Megatron-DeepSpeed | Bookcorpus +[Llama3-8B SFT](nlp/llm/llama3_8b/ColossalAI/applications/Colossal-LLaMA/README.md) | PyTorch | ColossalAI | school_math_0.25M +[Mamba-2](nlp/llm/mamba-2/megatron-lm/README.md) | PyTorch | Megatron-LM | GPT Small-117M +[Mixtral 8x7B](nlp/llm/mixtral/megatron-lm/README.md) | PyTorch | Megatron-LM | GPT Small-117M +[QWen-7B](nlp/llm/qwen-7b/firefly/README.md) | PyTorch | Firefly | qwen-7b +[QWen1.5-7B](nlp/llm/qwen1.5-7b/firefly/README.md) | PyTorch | Firefly | school_math +[QWen1.5-14B](nlp/llm/qwen1.5-14b/firefly/README.md) | PyTorch | Firefly | school_math +[Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/LLaMA-Factory/README.md) | PyTorch | LLaMA-Factory | qwen2.5-7b #### Text Correction -- Gitee