Skip to content

GitLab

"examples/dreambooth/train_dreambooth_lora.py" did not exist on "d87cc15977b87160c30abaace3894e802ad9e1e6"
Switch branch/tag
Find file Normal viewHistoryPermalink
README.md 4.34 KB
Newer Older
huchen's avatar
init the dlexamples new  
huchen committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
DCU深度学习样例
=====




# 介绍
基于PyTorch和TensorFlow深度学习框架在各个领域的常用测试样例

# Compute Vision
## Classification

| 框架 | 版本 | DCU | 精度 | 多DCU | 支持网络 | 代码位置
| :----------: | :----------: | :----------: | :----------: | :----------: | :----------: | :----------: |
unknown's avatar
修改首页README.md  
unknown committed
15
16
17 
| TensorFlow   | v1.15 | Yes | FP32/FP16 | Yes | ResNet50/VGG16/Inception-V3/MobileNet-V2 | [classfication](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow/ComputeVision/Classification) |
| TensorFlow   | v2.7 | Yes | FP32/FP16 | Yes | ResNet50/VGG16/Inception-V3 | [classfication](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/Classification) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | ResNet50/Vgg16/Inception-V3/MobileNet-V2 | [classfication](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Classification) |
huchen's avatar
init the dlexamples new  
huchen committed
18
19
20
21
22
23
24 

<br>

## Detection

| 框架 | 版本 | DCU | 精度 | 多DCU | 代码位置 |
| :----------: | :----------: | :----------: | :----------: | :----------: | :----------: |
unknown's avatar
修改首页README.md  
unknown committed
25
26
27
28
29
30
31
32
33 
| TensorFlow   | v2.7 | Yes | FP32/FP16 | Yes | [YOLO-V3](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/ComputeVision/Detection/YOLOv3) |
| TensorFlow   | v2.7 | Yes | FP32/FP16 | Yes | [Mask-Rcnn](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/ComputeVision/Detection/MaskRCNN) |
| TensorFlow   | v2.7 | Yes | FP32/FP16 | Yes | [SSD](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/ComputeVision/Detection/SSD) |
| PyTorch      | v1.10 | Yes | FP32 | Yes | [Faster-Rcnn](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Objection/Faster-rcnn) |
| PyTorch   | v1.10 | Yes | FP32/FP16 | Yes | [YOLO-V3](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Objection/YOLOv3) |
| PyTorch   | v1.10 | Yes | FP32/FP16 | Yes | [YOLO-V5](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Objection/yolov5) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [Mask-Rcnn](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Objection/MaskRCNN/vision) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [SSD](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Objection/ssd) |
huchen's avatar
init the dlexamples new  
huchen committed
34
35
36
37
38
39
40 

<br>

## Segmentation

| 框架 | 版本 | DCU | 精度 | 多DCU | 代码位置 |
| :----------: | :----------: | :----------: | :----------: | :----------: | :----------: |
unknown's avatar
修改首页README.md  
unknown committed
41 
| TensorFlow      | v2.7 | Yes | FP32/FP16 | Yes | [UNet](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/ComputeVision/Segmentation/UNet) |
huchen's avatar
init the dlexamples new  
huchen committed
42
43
44
45
46
47
48 

<br>

# NLP

| 框架 | 版本 |DCU | 精度 | 多DCU | 代码位置 |
| :----------: | :----------: | :----------: | :----------: | :----------: | :----------: |
unknown's avatar
修改首页README.md  
unknown committed
49
50
51
52
53
54 
| TensorFlow      | v1.15 | Yes | FP32/FP16 | Yes | [Bert](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow/NLP/Bert) |
| TensorFlow      | v2.7 | Yes | FP32/FP16 | Yes | [Bert](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/NLP/BERT/official) |
| TensorFlow      | v1.15 | Yes | FP32/FP16 | Yes | [Transformer](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow/NLP/Transformer_mlperf) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [Bert](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/NLP/BERT) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [GNMT](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/NLP/gnmt) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [Transformer](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/NLP/Transformer) |
huchen's avatar
init the dlexamples new  
huchen committed
55
56
57
58 


<br>
unknown's avatar
修改首页README.md  
unknown committed
59 
## Speech
huchen's avatar
init the dlexamples new  
huchen committed
60
61
62 

| 框架 | 版本 | DCU | 精度 | 多DCU | 代码位置 |
| :----------: | :----------: | :----------: | :----------: | :----------: | :----------: |
unknown's avatar
修改首页README.md  
unknown committed
63
64 
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [FastPitch](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Speech/FastPitch) |
| PyTorch      | v1.10 | Yes | FP32/FP16 | Yes | [Tacotron2](http://10.0.100.3/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Speech/Tacotron2) |
huchen's avatar
init the dlexamples new  
huchen committed
65
66
67 


<br>
huchen's avatar
Initial commit  
huchen committed
68