gilshm / mlperf-pytorch

ResNet-50 and MobileNet-v1 MLPerf models migrated to PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

mlperf-pytorch

ResNet-50 and MobileNet-v1 MLPerf models migrated to PyTorch.

Using the exact pre-processing stages used in MLPerf, ResNet-50 and MobileNet-v1 achieve top-1 accuracies of 76.456% and 71.676%, respectively.

Some words about the conversion process can be found here.

About

ResNet-50 and MobileNet-v1 MLPerf models migrated to PyTorch


Languages

Language:Python 100.0%