Optimize a ResNet50* V1.5 FP32 Training Model Package with TensorFlow*

Published: 10/23/2020  

Last Updated: 06/15/2022

Download Command

wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v2_3_0/resnet50v1-5-fp32-training.tar.gz

Description

This document has instructions for running ResNet50* v1.5 FP32 training using Intel® Optimization for TensorFlow*.

Note that the ImageNet dataset is used in these ResNet50 v1.5 examples. Download and preprocess the ImageNet dataset using the instructions here. After running the conversion script you should have a directory with the ImageNet dataset in the TF records format.

Quick Start Scripts

Script name Description
fp32_training_demo Launches a short run using small batch sizes and a limited number of steps to demonstrate the training flow
fp32_training_1_epoch Launches a test run that trains the model for one epoch and saves checkpoint files to an output directory.
fp32_training_full Trains the model using the full dataset and runs until convergence (90 epochs) and saves checkpoint files to an output directory. Note that this will take a considerable amount of time.
multi_instance_training_demo Uses numactl to launch one instance per socket of a short run using small batch sizes and a limited number of steps to demonstrate the training flow
multi_instance_training Uses numactl to launch one instance per socket for the full training flow. Checkpoint files and logs for each instance are saved to the output directory. Note that this will take a considerable amount of time.

Bare Metal

To run on bare metal, the following prerequisites must be installed in your environment:

Download and untar the model package and then run a quick start script.

DATASET_DIR=<path to the preprocessed imagenet dataset>
OUTPUT_DIR=<directory where checkpoint and log files will be written>

wget https://storage.googleapis.com/intel-optimized-tensorflow/models/v2_3_0/resnet50v1-5-fp32-training.tar.gz
tar -xvf resnet50v1-5-fp32-training.tar.gz
cd resnet50v1-5-fp32-training

quickstart/<script name>.sh

Documentation and Sources

Get Started​
Main GitHub*
Readme
Release Notes
Get Started Guide

Code Sources
Report Issue


License Agreement

LEGAL NOTICE: By accessing, downloading or using this software and any required dependent software (the “Software Package”), you agree to the terms and conditions of the software license agreements for the Software Package, which may also include notices, disclaimers, or license terms for third party software included with the Software Package. Please refer to the license file for additional details.


Related Containers and Solutions

ResNet50 V1.5 FP32 Training TensorFlow* Container

View All Containers and Solutions 🡢

Product and Performance Information

1

Performance varies by use, configuration and other factors. Learn more at www.Intel.com/PerformanceIndex.