Skip to content

Commit 9712cf8

Browse files
authored
Merge pull request NVIDIA#54 from vinhngx/master
adding resume capability for wget in case data downloads stop early
2 parents 9447242 + 8662db3 commit 9712cf8

2 files changed

Lines changed: 6 additions & 7 deletions

File tree

PyTorch/Segmentation/MaskRCNN/README.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Mask R-CNN is a convolution based neural network for the task of object instance
3232
The repository also contains scripts to interactively launch training, benchmarking and inference routines in a Docker container.
3333

3434
The major differences between the official implementation of the paper and our version of Mask R-CNN are as follows:
35-
- Mixed precision support with [PyTorch AMP](https://www.google.com/url?q=https://github.com/NVIDIA/apex&sa=D&ust=1552333999336000&usg=AFQjCNGO8dE0mQMO0lUfhMyCs7g0NmJXkw).
35+
- Mixed precision support with [PyTorch AMP](https://github.com/NVIDIA/apex).
3636
- Gradient accumulation to simulate larger batches.
3737
- Custom fused CUDA kernels for faster computations.
3838

@@ -111,8 +111,7 @@ This repository provides scripts to download and extract the COCO 2014 dataset.
111111

112112
To download, verify, and extract the COCO dataset, use the following scripts:
113113
```
114-
cd Detectron_PyT
115-
./download_dataset.sh <data/dir>
114+
../download_dataset.sh <data/dir>
116115
```
117116
By default, the data is organized into the following structure:
118117
```

PyTorch/Segmentation/MaskRCNN/download_dataset.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
DATA_DIR=$1
22

3-
wget https://dl.fbaipublicfiles.com/detectron/coco/coco_annotations_minival.tgz
4-
wget http://images.cocodataset.org/zips/train2014.zip
5-
wget http://images.cocodataset.org/zips/val2014.zip
6-
wget http://images.cocodataset.org/annotations/annotations_trainval2014.zip
3+
wget -c https://dl.fbaipublicfiles.com/detectron/coco/coco_annotations_minival.tgz
4+
wget -c http://images.cocodataset.org/zips/train2014.zip
5+
wget -c http://images.cocodataset.org/zips/val2014.zip
6+
wget -c http://images.cocodataset.org/annotations/annotations_trainval2014.zip
77

88
if md5sum -c hashes.md5
99
then

0 commit comments

Comments
 (0)