With a ResNet-50 backbone and a number of architectural modifications, this version provides better accuracy and performance.
The following section lists the requirements in order to start training the SSD300 v1.1 model.
Requirements
This repository contains Dockerfile which extends the PyTorch 22.10 NGC container
and encapsulates some dependencies. Aside from these dependencies,
ensure you have the following software:
- NVIDIA Docker
- PyTorch 22.10 NGC container
- GPU-based architecture:
For more information about how to get started with NGC containers, see the following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning Documentation:
- Getting Started Using NVIDIA GPU Cloud
- Accessing And Pulling From The NGC Container Registry
- Running PyTorch
For those unable to use the PyTorch 22.10 NGC container, to set up the required environment or create your own container, see the versioned NVIDIA Container Support Matrix.