search

mlcommons / training_results_v0.5

This repository contains the results and code for the MLPerf™ Training v0.5 benchmark.
https://mlcommons.org/en/training-normal-05/
Apache License 2.0
35 stars 54 forks source link

NVIDIA MLPerf v0.5.0 - Mask R-CNN pretrained RN50 backbone conversion #3

Closed cliffwoolley closed 5 years ago

cliffwoolley commented 5 years ago

Mask R-CNN conversion script to translate the reference Caffe2 pretrained RN50 backbone to PyTorch format.

Note that maskrcnn_benchmark/utils/c2_model_loading.py in the original NVIDIA submission already contained what's essentially this code (from https://github.com/facebookresearch/maskrcnn-benchmark/blob/optimized/maskrcnn_benchmark/utils/c2_model_loading.py). This PR merely brings in the minor tweak to that script that we had used to convert the model offline as opposed to online.

nvpaulius commented 5 years ago

merging given the approval above