Oneflow-Inc / OneFlow-Benchmark

OneFlow models for benchmarking.
104 stars 31 forks source link

OneFlow Deep Learning Benchmarks

Introduction

This repository provides OneFlow deep learning benchmark examples for CV, CTR and NLP, and more models are on the way and will be provided here when ready.

Convolutional Networks for Computer Vision Classification

Wide Deep Learning for Click-Through-Rate (CTR) Recommender Systems

BERT for Nature Language Process

GPT for Generative Pre-trained Transformer

OneFlow Benchmark Test Reports

Model DType XLA Throughput Speedup on 32 devices
ResNet50-V1.5 Float32 No 11.6k imges/sec 30.4
BERT base Pretrain Float32 No 530k tokens/sec 28.54